INNER CODE UNIT · Python
_get_oof_performance
AnotherSamWilson/miceforest · miceforest/imputation_kernel.py:515
def _get_oof_performance(
parameters: dict,
folds: Generator,
train_set: Dataset,
):
"""
Performance is gathered from built-in lightgbm.cv out of fold metric.
Optimal number of iterations is also obtained.
"""
num_iterations = parameters.pop("num_iterations")
lgbcv = cv(
params=parameters,
train_set=train_set,
folds=folds,
num_boost_round=num_iterations,
return_cvbooster=True,
callbacks=[