今天使用 sklearn.metrics.recall_score 计算 recall,遇到了这个报错,官网说明如下:
第一反应没太看懂,根据别人的改法把 pos_label 改成 ‘neg’ 之类的也没用。
然后突然灵机一动,pos_label 应该就是 positive label 的意思,也就是正类,试了试,确实如此,记录一下这个小小的发现。
使用函数计算 precision 和 f1-score 也是一样的道理。
代码示意如下:
recall_score(ground_truth,prediction,pos_label='正类(数字的话不需要引号)')