第57集 python机器学习:混淆矩阵( 三 )

二分类混淆矩阵

下面我们用混淆矩阵来比较前面拟合过的模型(两个虚拟模型、决策树和Logistic回归):

print(\"Most frequency class:\")

print(confusion_matrix(y_test pred_most_frequent))

print(\"\\nDummy model:\")

print(confusion_matrix(y_test pred_dummy))

print(\"\\nDecision tree:\")

print(confusion_matrix(y_test pred_tree))

print(\"\\nlogistic regression\")

print(confusion_matrix(y_test pred_logreg))

打印出来的结果为:

Most frequency class:

[[403   0

推荐阅读