from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt

y_test = [0, 0, 0, 0, 0, 1, 1, 1, 1, 1]
y_pred = [0, 1, 0, 0, 0, 0, 0, 1, 1, 1]
confusion_matrix = confusion_matrix(y_test, y_pred)
print(confusion_matrix)
plt.matshow(confusion_matrix)
plt.title(\'Confusion matrix\')
plt.colorbar()
plt.ylabel(\'True label\')
plt.xlabel(\'Predicted label\')
plt.show()
[[4 1]
 [2 3]]








							

   
版权声明：本文为starcrm原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。



本文链接：https://www.cnblogs.com/starcrm/p/11718616.html