我正在做一个多级分类问题,目标标签为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'])
从左到右给你"狗","鸟","鱼"。