From 4500e32fbe4eb083ba1140806a62c482a6a06770 Mon Sep 17 00:00:00 2001
From: Sefik Ilkin Serengil <serengil@gmail.com>
Date: Sat, 26 Mar 2022 08:25:36 +0000
Subject: [PATCH] ensemble bug

---
 tests/Ensemble-Face-Recognition.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/tests/Ensemble-Face-Recognition.py b/tests/Ensemble-Face-Recognition.py
index 3e8c491..3e176b4 100644
--- a/tests/Ensemble-Face-Recognition.py
+++ b/tests/Ensemble-Face-Recognition.py
@@ -45,6 +45,8 @@ for key, values in idendities.items():
 positives = pd.DataFrame(positives, columns = ["file_x", "file_y"])
 positives["decision"] = "Yes"
 
+positives = positives.sample(12)
+
 print(positives.shape)
 #--------------------------
 #Negatives
@@ -206,17 +208,13 @@ gbm.save_model("face-recognition-ensemble-model.txt")
 #--------------------------
 #Evaluation
 
-prediction_classes = gbm.predict(x_test)
+predictions = gbm.predict(x_test)
 
 prediction_classes = []
 for prediction in predictions:
     prediction_class = np.argmax(prediction)
     prediction_classes.append(prediction_class)
 
-print(y_test)
-print("------")
-print(prediction_classes)
-
 cm = confusion_matrix(y_test, prediction_classes)
 print(cm)