abu
abu

Reputation: 777

Cross-validation predictions for lightGBM

Is there a simple way to recover cross-validation predictions from the model built using lgb.cv from lightGBM?

I am doing a grid search combined with cross validation. Ultimately I would like to obtain the predictions for each of the defined hold-out folds so I can also stack a few models.

Upvotes: 2

Views: 1819

Answers (1)

norm
norm

Reputation: 339

the only way i have found is to continue to train (lgb.train) the same model (booster) to do this you must use init_model='model.txt'. Then save the models best iteration like this bst.save_model('model.txt', num_iteration=bst.best_iteration). Note this is python api, sorry. I have asked the same question but for the python api

Upvotes: 1

Related Questions