From db13fbf8d7b9446fd19eb0feed017c0e453e3421 Mon Sep 17 00:00:00 2001
From: Sefik Ilkin Serengil <serengil@gmail.com>
Date: Wed, 23 Mar 2022 20:37:01 +0000
Subject: [PATCH] prediction_classes

---
 tests/Ensemble-Face-Recognition.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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
+#--------------------------