您的位置:首页 > 其它

sklearn.metrics.precision_score 中 unknow is not supported 问题

2017-08-10 10:47 260 查看
最近用 XGBoost 对某些东西分类,类别用 1,2,3,4,5,6,7 来表示。

然后交叉验证出了 test_y 和 preds 集合,使用 precision_score() 函数缺报错,提示

unknown is not supported

经过研究,发现类别是用 object 类型表示的,而 precision_score() 需要用 int 来表示,改正:

test_y = np.array(test_y.astype(int))
preds = preds.astype(int)


即可。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐