【发布时间】:2019-09-14 08:17:41
【问题描述】:
试图获得决策树的混淆矩阵
tn, fp, fn, tp = confusion_matrix(y_test, y_pred_tree)
print(tn, fp, fn, tp)
confusion_matrix(y_test, y_pred_tree)
ValueError Traceback(最近一次调用最后一次) 在 () ---->
1 tn, fp, fn, tp = 混淆矩阵(y_test, y_pred_tree)
2 打印(tn,fp,fn,tp)
3
4 混淆矩阵(y_test, y_pred_tree)ValueError:要解压的值太多(预计为 4 个)
【问题讨论】:
-
也许你应该尝试打印
confusion_matrix的输出(不拆包),看看它的形状是什么。 (即print(confusion_matrix(y_test, y_pred_tree)) -
你在使用
sklearn.metrics.confusion_matrix吗?如果是这样,tn, fp, fn, tp = confusion_matrix(y_test, y_pred_tree)将不起作用。该函数返回一个带有混淆矩阵的array对象。试试print(confusion_matrix(y_test, y_pred_tree))