Commit 196bf7a1 authored by WuNoah's avatar WuNoah
Browse files

change

parent ada0db43
...@@ -83,7 +83,7 @@ class ScriptClassifierMerge(ScriptClassifier): ...@@ -83,7 +83,7 @@ class ScriptClassifierMerge(ScriptClassifier):
confusionMatrix = np.transpose(confusion_matrix(test_y, pred_y)) confusionMatrix = np.transpose(confusion_matrix(test_y, pred_y))
# in case no success individus and confusion matrix will be [[x]] # in case no success individus and confusion matrix will be [[x]]
if (len(confusionMatrix) == 1): if (len(confusionMatrix) == 0):
confusionMatrix = np.array([[0, 0], [0, confusionMatrix[0][0]]]) confusionMatrix = np.array([[0, 0], [0, confusionMatrix[0][0]]])
accuracy = accuracy_score(test_y, pred_y) accuracy = accuracy_score(test_y, pred_y)
......
Supports Markdown
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