我正在进行多类分类问题,目标标签为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)告诉我如何检查订购?必须有一些参数,我的标签订购存储在哪里?如果我的标签是"狗","鸟","鱼"?它会按字母顺序排列吗?
答案 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'])
从左到右给你'狗','鸟','鱼'。