xgboost4j报错:Check failed:preds.Size()==info.labels_.Size()

在使用xgboost4j-spark-0.90版本时遇到错误:Check failed: preds.Size()==info.labels_.Size()。错误源于配置参数设置。当objective设定为'binary:logistic'时,无需设置'num_class'。若为多分类问题,'num_class'则必须指定。博主建议使用最新版xgboost4j以避免类似问题。
摘要由CSDN通过智能技术生成

使用的xgboost4j的版本是:xgboost4j-spark-0.90。
运行过程中报出以下错误:
Check failed:preds.Size()==info.labels_.Size()
在这里插入图片描述

报错发生原因不清楚。
解决方案是:
xgboost的配置参数里,如果指定了
"objective" -> "binary:logistic"
"num_class" -> 2无须设置,就是把这个参数注释掉。
这样子错误就解决了。
但如果"objective"设置了多分类,则"num_class"这个参数必须设置。

感觉xgboost4j有很多的坑,大家尽量用最新版的。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值