拆分数据集 X_train, X_test, y_train, y_test = train_test_split(heartbeats_image, labels, test_size=0.2, random_state=42) X_train, X_val, y_train, y_val = train_test_split(X_train, y_train, test_size=0.2, random_state=42) # 保存数据集 np.save('X_train.npy', X_train) np.save('X_val.npy', X_val) np.save('X_test.npy', X_test) np.save('y_train.npy', y_train) np.save('y_val.npy', y_val) np.save('y_test.npy', y_test) from keras.models import Sequential from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout # 定义卷积神经网络 model = Sequential([ Conv2D(filters=32, kernel_size=(3,3), activation='relu', input_shape=(255,255,1)), MaxPooling2D(pool_size=(2,2)), Conv2D(filters=64, kernel_size=(3,3), activation='relu'), MaxPooling2D(pool_size=(2,2)), Conv2D(filters=128, kernel_size=(3,3), activation='relu'), MaxPooling2D(pool_size=(2,2)), Flatten(), Dense(units=128, activation='relu'), Dropout(0.5), Dense(units=1, activation='sigmoid') ]) model.add(Dense(20, activation='softmax')) # 编译模型 model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) # 训练模型 history = model.fit(X_train, y_train, epochs=10, validation_data=(X_val, y_val)) # 保存模型 model.save('my_model.h5') from sklearn.metrics import confusion_matrix, roc_curve, auc import matplotlib.pyplot as plt # 对测试集进行预测 y_pred = model.predict(X_test) # 将预测结果转换为标签 y_pred_labels = (y_pred > 0.5).astype(int) from sklearn.metrics import confusion_matrix from sklearn.utils.multiclass import unique_labels # 将多标签指示器转换成标签数组 y_test = unique_labels(y_test) y_pred_labels = unique_labels(y_pred_labels) # 计算混淆矩阵 cm = confusion_matrix(y_test, y_pred_labels) # 绘制混淆矩阵 plt.imshow(cm, cmap=plt.cm.Blues) plt.xlabel("Predicted labels") plt.ylabel("True labels") plt.xticks([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19], ['N','L','R','A','a','J','S','V','F','[','!',']','e','j','E','/','f','x','Q','|']) plt.yticks([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19], ['N','L','R','A','a','J','S','V','F','[','!',']','e','j','E','/','f','x','Q','|']) plt.title('Confusion matrix') plt.colorbar() plt.show()之后怎么绘制ROC曲线，let's think step by step

首先需要计算出 ROC 曲线上的真正率 (True Positive Rate, TPR) 和假正率 (False Positive Rate, FPR)。可以使用 scikit-learn 库中的 roc_curve 函数来计算。

代码如下：

fpr, tpr, thresholds = roc_curve(y_test, y_pred)

其中，y_test 是测试集的真实标签，y_pred 是模型预测出的标签。

然后，可以使用 auc 函数计算曲线下面积（Area Under Curve, AUC）。

代码如下：

roc_auc = auc(fpr, tpr)

最后，使用 matplotlib 库绘制 ROC 曲线。

代码如下：

plt.plot(fpr, tpr, color='darkorange', label='ROC curve (area = %0.2f)' % roc_auc)
plt.plot([0, 1], [0, 1], color='navy', linestyle='--')
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 1.05])
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('Receiver operating characteristic')
plt.legend(loc="lower right")
plt.show()

其中，darkorange 是曲线的颜色，navy 是虚线的颜色，label 是曲线的标签，xlim 和 ylim 分别设置 X 轴和 Y 轴的范围，xlabel 和 ylabel 分别设置 X 轴和 Y 轴的标签，title 设置曲线的标题，legend 设置曲线的图例。