from sklearn.metrics import confusion_matrix # 导入混淆矩阵 cm = confusion_matrix(y_test, y_pred) # 调用混淆矩阵 plt.title("ANN Confusion Matrix") # 标题:人工神经网络混淆矩阵 sns.heatmap(cm, annot=True, cmap="Blues", fmt="d", cbar=False) # 热力图设定 plt.show() # 显示混淆矩阵
