Как узнать порядок актуальных этикеток для Confusion Matrix?

Question

Я запутался в том, как узнать настоящие ярлыки в Confusion Matrix? Я знаю, чтобы передать метки, но мой главный вопрос - как мы узнаем, какую последовательность я должен передать метке?

from sklearn.metrics import confusion_matrix

cm = confusion_matrix(y_test,y_pred_classes)

Это возвращает результат функции confusion_matrix ():

Затем я объявил метки и передал метки для построения матрицы путаницы:

import itertools

def plotConfusionMatrix(cm, classes, normalize=False, title='Confusion Matrix', cmap = plt.cm.Blues):

    plt.figure(figsize = (10,7))
    plt.imshow(cm, interpolation='nearest', cmap=cmap)
    plt.title(title)
    plt.colorbar()
    tick_marks = np.arange(len(classes))
    plt.xticks(tick_marks, classes, rotation=45)
    plt.yticks(tick_marks, classes)

    if normalize:
        cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]
        print('Normalized Confusion Matrix')
    else:
        print('Un-normalized Confusion Matrix')

    print(cm)

    thresh = cm.max()/2

    for i,j in itertools.product(range(cm.shape[0]), range(cm.shape[1])):
        plt.text(j,i, cm[i,j], horizontalalignment='center', color='white' if cm[i,j] > thresh else 'black', fontsize=25, fontweight='bold')
        plt.tight_layout()
        plt.ylabel('Actual Class')
        plt.xlabel('Predicted Class')

Затем вызвал функцию и передал метки:

classes = ['climbingdown','climbingup','jumping','lying','running','sitting','standing','walking']
plotConfusionMatrix(cm, classes)

Вывод для построенной матрицы путаницы был:

Теперь мой точный вопрос: я прошел метки каждого класса, но как я доберусь до знаете порядок, в котором я должен пройти?

Answer 1

Вы можете передать метки классов в функцию матрицы смешения .

Если вы этого не сделаете, он просто будет использовать отсортированный порядок меток. Так что я думаю, это зависит от того, как отображаются ваши ярлыки y_true и y_pred.