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

来源:互联网 发布:excel表格怎么数据统计 编辑:程序博客网 时间:2024/06/05 15:04

最近用 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)

即可。

原创粉丝点击