Skip to content

Commit 4feee28

Browse files
[python-package] Rename cvfolds variable into cvbooster in engine.cv function (#6762)
1 parent 480600b commit 4feee28

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

python-package/lightgbm/engine.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -758,7 +758,7 @@ def cv(
758758
train_set._update_params(params)._set_predictor(predictor)
759759

760760
results = defaultdict(list)
761-
cvfolds = _make_n_folds(
761+
cvbooster = _make_n_folds(
762762
full_data=train_set,
763763
folds=folds,
764764
nfold=nfold,
@@ -802,24 +802,24 @@ def cv(
802802
for cb in callbacks_before_iter:
803803
cb(
804804
callback.CallbackEnv(
805-
model=cvfolds,
805+
model=cvbooster,
806806
params=params,
807807
iteration=i,
808808
begin_iteration=0,
809809
end_iteration=num_boost_round,
810810
evaluation_result_list=None,
811811
)
812812
)
813-
cvfolds.update(fobj=fobj) # type: ignore[call-arg]
814-
res = _agg_cv_result(cvfolds.eval_valid(feval)) # type: ignore[call-arg]
813+
cvbooster.update(fobj=fobj) # type: ignore[call-arg]
814+
res = _agg_cv_result(cvbooster.eval_valid(feval)) # type: ignore[call-arg]
815815
for _, key, mean, _, std in res:
816816
results[f"{key}-mean"].append(mean)
817817
results[f"{key}-stdv"].append(std)
818818
try:
819819
for cb in callbacks_after_iter:
820820
cb(
821821
callback.CallbackEnv(
822-
model=cvfolds,
822+
model=cvbooster,
823823
params=params,
824824
iteration=i,
825825
begin_iteration=0,
@@ -828,14 +828,14 @@ def cv(
828828
)
829829
)
830830
except callback.EarlyStopException as earlyStopException:
831-
cvfolds.best_iteration = earlyStopException.best_iteration + 1
832-
for bst in cvfolds.boosters:
833-
bst.best_iteration = cvfolds.best_iteration
831+
cvbooster.best_iteration = earlyStopException.best_iteration + 1
832+
for bst in cvbooster.boosters:
833+
bst.best_iteration = cvbooster.best_iteration
834834
for k in results:
835-
results[k] = results[k][: cvfolds.best_iteration]
835+
results[k] = results[k][: cvbooster.best_iteration]
836836
break
837837

838838
if return_cvbooster:
839-
results["cvbooster"] = cvfolds # type: ignore[assignment]
839+
results["cvbooster"] = cvbooster # type: ignore[assignment]
840840

841841
return dict(results)

0 commit comments

Comments
 (0)