Как упорядочены выходы precision_recall_fscore_support в sklearn?

Question

Я делаю многоклассовую классификационную задачу с целевыми метками 0, 1 и 2 (dtype=int). Я построил свою модель и хочу оценить, насколько она хороша, используя точность, отзыв и f-рейтинг. Вот что я сделал:

precision,recall,fscore,support = precision_recall_fscore_support(y_train,y_train_pred)
print('recall:  {0}'.format(recall))

И вот мой вывод:

recall:  [ 0.99991709  0.56711409  0.12327412]

Я почти уверен, что порядок составляет 0, 1, 2 слева направо. Но я не знаю точно. Кто-нибудь, пожалуйста, (1) подтвердит заказ для меня и (2) подскажет, как я могу проверить заказ? Должны быть какие-то параметры, где хранится мой порядок меток? Что если бы мои ярлыки были "собака", "птица", "рыба"? Было бы это по алфавиту?

Answer 1

precision,recall,fscore,support = precision_recall_fscore_support(y_train,y_train_pred,labels=[0,1,2])

Дает вам 0, 1, 2 слева направо. Как

precision,recall,fscore,support = precision_recall_fscore_support(y_train,y_train_pred,labels=['dog','bird','fish'])

Дает вам "собака", "птица", "рыба" слева направо.