diff --git a/demo.py b/demo.py index aef02d3..78335ae 100644 --- a/demo.py +++ b/demo.py @@ -21,7 +21,7 @@ #Let Genetic Programming find best ML model and hyperparameters tpot = TPOTClassifier(generations=5, verbosity=2) tpot.fit(tele.drop('Class', axis=1).loc[training_indices].values, - tele.loc[training_indicss, 'Class'].values) + tele.loc[training_indices, 'Class'].values) #Score the accuracy tpot.score(tele.drop('Class', axis=1).loc[validation_indices].values,