抱歉，上述代码中的 plt.colorbar() 出现了错误。这是因为在混淆矩阵的情况下，不需要显示颜色条例。您可以将该行代码从代码中删除，并重新运行：
import numpy as np
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
# 计算混淆矩阵
cm = confusion_matrix(y_test, lr_yhat)
# 创建标签数组
labels = ['Non-Fraud', 'Fraud']
# 绘制混淆矩阵图像
disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=labels)
disp.plot(cmap='Blues')
# 设置标题和轴标签
plt.title('Confusion Matrix')
plt.xlabel('Predicted Label')
plt.ylabel('True Label')
# 展示图像
plt.show()
这样就能够正确绘制出混淆矩阵图像。
内容由零声教学AI助手提供，问题来源于学员提问
