prediction_classes

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

View File

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