diff --git a/tests/Ensemble-Face-Recognition.py b/tests/Ensemble-Face-Recognition.py index 3d5a688..144be36 100644 --- a/tests/Ensemble-Face-Recognition.py +++ b/tests/Ensemble-Face-Recognition.py @@ -202,7 +202,7 @@ gbm.save_model("face-recognition-ensemble-model.txt") #-------------------------- #Evaluation -predictions = gbm.predict(x_test) +prediction_classes = gbm.predict(x_test) cm = confusion_matrix(y_test, prediction_classes) print(cm) @@ -246,4 +246,4 @@ auc = metrics.roc_auc_score(y_test, y_pred_proba) plt.figure(figsize=(7,3)) plt.plot(fpr,tpr,label="data 1, auc="+str(auc)) -#-------------------------- \ No newline at end of file +#--------------------------