如何在sklearn中对precision_recall_fscore_support的输出进行排序?

时间:2017-12-08 00:45:34

标签: python-3.x scikit-learn precision-recall multiclass-classification

我正在进行多类分类问题,目标标签为0,1和2(dtype = int)。我构建了我的模型,想要评估使用精度,召回和f-score的好处。这就是我的所作所为:

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)告诉我如何检查订购?必须有一些参数,我的标签订购存储在哪里?如果我的标签是"狗","鸟","鱼"?它会按字母顺序排列吗?

1 个答案:

答案 0 :(得分: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'])

从左到右给你'狗','鸟','鱼'。