如何替换混乱矩阵的行和列?

时间:2019-03-27 08:48:44

标签: python confusion-matrix

我有一个代码试图对船级进行分类。我已经完成并提取了混淆矩阵。但是现在我想安排矩阵,这样每个人都会看到同一类的序列而不是字母。由于缺乏声誉,我无法加载混乱矩阵的图片。

我要替换标签及其指定的标签值: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()

0 个答案:

没有答案