精确回忆曲线,平均值='微'用于scikit-learn中的多类分类器

时间:2014-12-18 08:13:53

标签: python scikit-learn classification

我正在做的就是在此页面上运行提供的代码:http://scikit-learn.org/stable/auto_examples/plot_precision_recall.html 找到ROC曲线。

我所做的就是复制代码,但是我收到了这个错误:

TypeError: average_precision_score() got an unexpected keyword argument 'average'

我该怎么做才能删除错误?我想将它用于多类情况,所以我确实希望保持“average ='micro'”部分。

1 个答案:

答案 0 :(得分:2)

版本0.15之前average参数不存在,所以我认为你必须有一个旧版本的scikit-learn。请参阅change log

“多标签指标格式的多标签分类输出现在由Arnaud Joly的metrics.roc_auc_score和metrics.average_precision_score支持。”

代码在0.15.2下运行。请参阅第二个答案here以在运行时检查python模块的版本。