Skip to content

R
robo-dividend
Commit 1dc3eb3a authored by wenwen.tang's avatar wenwen.tang 😕
Browse files
Options

bugfix 二维矩阵

parent c0a487e0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ai/model_trainer.py
View file @ 1dc3eb3a
...@@ -23,7 +23,7 @@ class ModelTrainer(ABC): ...@@ -23,7 +23,7 @@ class ModelTrainer(ABC):
print(strMethod + " ====== test results ======") print(strMethod + " ====== test results ======")
y_pred = classifier.predict(X_test) 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(strMethod + " Confusion Matrix:")
print(result0) print(result0)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment