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

prediction_classes

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2022-03-23 20:37:01 +00:00 committed by GitHub
parent 466037ef5e
commit db13fbf8d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -202,7 +202,7 @@ gbm.save_model("face-recognition-ensemble-model.txt")
#-------------------------- #--------------------------
#Evaluation #Evaluation
predictions = gbm.predict(x_test) prediction_classes = gbm.predict(x_test)
cm = confusion_matrix(y_test, prediction_classes) cm = confusion_matrix(y_test, prediction_classes)
print(cm) print(cm)
@ -246,4 +246,4 @@ auc = metrics.roc_auc_score(y_test, y_pred_proba)
plt.figure(figsize=(7,3)) plt.figure(figsize=(7,3))
plt.plot(fpr,tpr,label="data 1, auc="+str(auc)) plt.plot(fpr,tpr,label="data 1, auc="+str(auc))
#-------------------------- #--------------------------