使用的xgboost4j的版本是:xgboost4j-spark-0.90。
运行过程中报出以下错误:
Check failed:preds.Size()==info.labels_.Size()
报错发生原因不清楚。
解决方案是:
xgboost的配置参数里,如果指定了
"objective" -> "binary:logistic"
则 "num_class" -> 2
无须设置,就是把这个参数注释掉。
这样子错误就解决了。
但如果"objective"
设置了多分类,则"num_class"
这个参数必须设置。
感觉xgboost4j有很多的坑,大家尽量用最新版的。