How to get the booster from crossvalidation in xgboost4j


#1

Hi,

I am using xgboost4j. The crossValidation() function just returns an array of string. Hence how to get the booster after the crossvalidation?


#2

val cv = new CrossValidator()
val cvModel = cv.fit(dataset)
val bestModel = cvModel.bestModel.asInstanceOf[XGBoostModelType]