我有一个代码试图对船级进行分类。我已经完成并提取了混淆矩阵。但是现在我想安排矩阵,这样每个人都会看到同一类的序列而不是字母。由于缺乏声誉,我无法加载混乱矩阵的图片。
我要替换标签及其指定的标签值:Ada, Burak, Doğan, Gabya, Kılıç, Meko, Rüzgar, Yıldız, YTKB
作为行和列
为此:Gabya, Meko, Ada, Burak, YTKB, Kılıç, Doğan, Rüzgar, Yıldız
作为行和标签。
def plot_confusion_matrix(y_true, y_pred, classes,
normalize=True,
title=None,
cmap=plt.cm.Reds):
cm = confusion_matrix(y_test, out)
cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]
fig, ax = plt.subplots()
im = ax.imshow(cm, interpolation='nearest', cmap=cmap)
ax.figure.colorbar(im, ax=ax)
ax.set(xticks=np.arange(cm.shape[1]),
yticks=np.arange(cm.shape[0]),
xticklabels=classes,
yticklabels=classes,
title=title)
plt.setp(ax.get_xticklabels(), rotation=45, ha="right",
rotation_mode="anchor")
plot_confusion_matrix(y_test, out, classes=classes, normalize=True)
plt.imshow()