github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 19:45:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

ensemble bug

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2022-03-26 08:25:36 +00:00
parent 527a427443
commit 4500e32fbe
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/Ensemble-Face-Recognition.py
View File

@ -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)