Unverified Commit 4feee28a authored by Claudio Salvatore Arcidiacono's avatar Claudio Salvatore Arcidiacono Committed by GitHub
Browse files

[python-package] Rename `cvfolds` variable into `cvbooster` in `engine.cv` function (#6762)

parent 480600b3
...@@ -758,7 +758,7 @@ def cv( ...@@ -758,7 +758,7 @@ def cv(
train_set._update_params(params)._set_predictor(predictor) train_set._update_params(params)._set_predictor(predictor)
results = defaultdict(list) results = defaultdict(list)
cvfolds = _make_n_folds( cvbooster = _make_n_folds(
full_data=train_set, full_data=train_set,
folds=folds, folds=folds,
nfold=nfold, nfold=nfold,
...@@ -802,7 +802,7 @@ def cv( ...@@ -802,7 +802,7 @@ def cv(
for cb in callbacks_before_iter: for cb in callbacks_before_iter:
cb( cb(
callback.CallbackEnv( callback.CallbackEnv(
model=cvfolds, model=cvbooster,
params=params, params=params,
iteration=i, iteration=i,
begin_iteration=0, begin_iteration=0,
...@@ -810,8 +810,8 @@ def cv( ...@@ -810,8 +810,8 @@ def cv(
evaluation_result_list=None, evaluation_result_list=None,
) )
) )
cvfolds.update(fobj=fobj) # type: ignore[call-arg] cvbooster.update(fobj=fobj) # type: ignore[call-arg]
res = _agg_cv_result(cvfolds.eval_valid(feval)) # type: ignore[call-arg] res = _agg_cv_result(cvbooster.eval_valid(feval)) # type: ignore[call-arg]
for _, key, mean, _, std in res: for _, key, mean, _, std in res:
results[f"{key}-mean"].append(mean) results[f"{key}-mean"].append(mean)
results[f"{key}-stdv"].append(std) results[f"{key}-stdv"].append(std)
...@@ -819,7 +819,7 @@ def cv( ...@@ -819,7 +819,7 @@ def cv(
for cb in callbacks_after_iter: for cb in callbacks_after_iter:
cb( cb(
callback.CallbackEnv( callback.CallbackEnv(
model=cvfolds, model=cvbooster,
params=params, params=params,
iteration=i, iteration=i,
begin_iteration=0, begin_iteration=0,
...@@ -828,14 +828,14 @@ def cv( ...@@ -828,14 +828,14 @@ def cv(
) )
) )
except callback.EarlyStopException as earlyStopException: except callback.EarlyStopException as earlyStopException:
cvfolds.best_iteration = earlyStopException.best_iteration + 1 cvbooster.best_iteration = earlyStopException.best_iteration + 1
for bst in cvfolds.boosters: for bst in cvbooster.boosters:
bst.best_iteration = cvfolds.best_iteration bst.best_iteration = cvbooster.best_iteration
for k in results: for k in results:
results[k] = results[k][: cvfolds.best_iteration] results[k] = results[k][: cvbooster.best_iteration]
break break
if return_cvbooster: if return_cvbooster:
results["cvbooster"] = cvfolds # type: ignore[assignment] results["cvbooster"] = cvbooster # type: ignore[assignment]
return dict(results) return dict(results)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment