是否可以在不同数量的树上保存xgboost模型?

kjthegm6  于 2021-05-27  发布在  Spark
关注(0)|答案(0)|浏览(358)

我是xgboost的新手。我正在运行xgboost4jspark0.7来训练带有sparkDataframe的logistic模型。例如,我正在和

xgbModelWithDF = XGBoost.trainWithDataFrame()

保存一个经过训练的模型

xgbModelWithDF.booster.savemodel("file_path")

并进行批量预测

xgbModelWithDF.transform(testdata)

我的问题是,如果我需要分别保存round=500、1000、1500、2000(树数)的模型怎么办。我可以增量运行吗?或者我可以运行round=2000并在任何特定的中间点保存任何模型吗?如果是,请告知。
短暂性脑缺血发作

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题