是否有一种简单的方法可以从lgb.cv
使用lightGBM
构建的模型中恢复交叉验证预测?
我正在进行网格搜索并结合交叉验证。最后,我想获得每个定义的保持折叠的预测,这样我也可以叠加几个模型。
答案 0 :(得分:1)
我找到的唯一方法是继续训练(lgb.train
)相同的模型(助推器),你必须使用init_model='model.txt'
。然后保存模型最佳迭代,如bst.save_model('model.txt', num_iteration=bst.best_iteration)
。注意这是python api,对不起。我问了同样的question,但是对于python api