python-3.x - sklearn 中precision_recall_fscore_support 的输出是如何排序的？

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'])

从左到右为您提供“狗”、“鸟”、“鱼”。