lightGBM的交叉验证预测

时间:2018-01-25 10:10:14

标签: r cross-validation grid-search ensemble-learning lightgbm

是否有一种简单的方法可以从lgb.cv使用lightGBM构建的模型中恢复交叉验证预测?

我正在进行网格搜索并结合交叉验证。最后,我想获得每个定义的保持折叠的预测,这样我也可以叠加几个模型。

1 个答案:

答案 0 :(得分:1)

我找到的唯一方法是继续训练(lgb.train)相同的模型(助推器),你必须使用init_model='model.txt'。然后保存模型最佳迭代,如bst.save_model('model.txt', num_iteration=bst.best_iteration)。注意这是python api,对不起。我问了同样的question,但是对于python api