Commit 25330637 authored by WuNoah's avatar WuNoah
Browse files

fix bug

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