precision_recall_fscore_support的输出如何有序



我正在做一个多级分类问题,目标标签为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)告诉我如何检查订单?必须有一些参数存储我的标签排序?如果我的标签是"狗","鸟","鱼"怎么办?它会是字母表吗?

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

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

相关内容

  • 没有找到相关文章

最新更新