Skip to content

Commit e7f56b7

Browse files
authored
fix UndefVarError in update method (#28)
1 parent 3fbbafe commit e7f56b7

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/MLJCatBoostInterface.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ function MMI.update(mlj_model::CatBoostModels, verbosity::Integer, fitresult, ca
144144
if mlj_model.iterations > cache.mlj_model.iterations &&
145145
MMI.is_same_except(mlj_model, cache.mlj_model, :iterations)
146146
iterations = mlj_model.iterations - cache.mlj_model.iterations
147+
verbose = verbosity <= 1 ? false : true
147148
new_model = model_init(mlj_model; verbose, iterations)
148149
new_model.fit(data_pool; init_model=fitresult)
149150
report = (feature_importances=feature_importance(new_model),)

0 commit comments

Comments
 (0)