写出代码要求,在jupyter notebook中将预测分类的结果使用混淆矩阵做出可视化的程序
时间: 2023-03-10 19:16:51 浏览: 160
HurleyWong#TechNote#混淆矩阵可视化1
答:代码如下:import pandas as pd
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix# 读取预测结果
prediction = pd.read_csv('prediction.csv')# 生成混淆矩阵
cm = confusion_matrix(prediction['true'], prediction['predict'])# 可视化
plt.matshow(cm, cmap=plt.cm.Blues)
plt.title('Confusion Matrix')
plt.xlabel('Predicted')
plt.ylabel('True')
plt.xticks([0, 1], ['0', '1'])
plt.yticks([0, 1], ['0', '1'])# 添加文本
for i in range(2):
for j in range(2):
plt.text(j, i, cm[i, j], ha='center', va='center', color='red')# 显示
plt.show()
阅读全文