@@ -758,7 +758,7 @@ def cv(
758
758
train_set ._update_params (params )._set_predictor (predictor )
759
759
760
760
results = defaultdict (list )
761
- cvfolds = _make_n_folds (
761
+ cvbooster = _make_n_folds (
762
762
full_data = train_set ,
763
763
folds = folds ,
764
764
nfold = nfold ,
@@ -802,24 +802,24 @@ def cv(
802
802
for cb in callbacks_before_iter :
803
803
cb (
804
804
callback .CallbackEnv (
805
- model = cvfolds ,
805
+ model = cvbooster ,
806
806
params = params ,
807
807
iteration = i ,
808
808
begin_iteration = 0 ,
809
809
end_iteration = num_boost_round ,
810
810
evaluation_result_list = None ,
811
811
)
812
812
)
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]
815
815
for _ , key , mean , _ , std in res :
816
816
results [f"{ key } -mean" ].append (mean )
817
817
results [f"{ key } -stdv" ].append (std )
818
818
try :
819
819
for cb in callbacks_after_iter :
820
820
cb (
821
821
callback .CallbackEnv (
822
- model = cvfolds ,
822
+ model = cvbooster ,
823
823
params = params ,
824
824
iteration = i ,
825
825
begin_iteration = 0 ,
@@ -828,14 +828,14 @@ def cv(
828
828
)
829
829
)
830
830
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
834
834
for k in results :
835
- results [k ] = results [k ][: cvfolds .best_iteration ]
835
+ results [k ] = results [k ][: cvbooster .best_iteration ]
836
836
break
837
837
838
838
if return_cvbooster :
839
- results ["cvbooster" ] = cvfolds # type: ignore[assignment]
839
+ results ["cvbooster" ] = cvbooster # type: ignore[assignment]
840
840
841
841
return dict (results )
0 commit comments