bugfix 二维矩阵

1dc3eb3a · wenwen.tang · c0a487e0 · 1dc3eb3a
Commit 1dc3eb3a authored Dec 12, 2023 by wenwen.tang 😕
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

model_trainer.py ai/model_trainer.py +1 -1

No files found.
--- a/ai/model_trainer.py
+++ b/ai/model_trainer.py
@@ -23,7 +23,7 @@ class ModelTrainer(ABC):
        print(strMethod + " ====== test results ======")
        y_pred = classifier.predict(X_test)

-        result0 = confusion_matrix(y_test, y_pred)
+        result0 = confusion_matrix(y_test, y_pred, labels=[0, 1])
        print(strMethod + " Confusion Matrix:")
        print(result0)