fix UndefVarError in update method (#28)

JuliaAI · Jul 25, 2023 · e7f56b7 · e7f56b7
1 parent 3fbbafe
commit e7f56b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/MLJCatBoostInterface.jl b/src/MLJCatBoostInterface.jl
@@ -144,6 +144,7 @@ function MMI.update(mlj_model::CatBoostModels, verbosity::Integer, fitresult, ca
     if mlj_model.iterations > cache.mlj_model.iterations &&
        MMI.is_same_except(mlj_model, cache.mlj_model, :iterations)
         iterations = mlj_model.iterations - cache.mlj_model.iterations
+        verbose = verbosity <= 1 ? false : true
         new_model = model_init(mlj_model; verbose, iterations)
         new_model.fit(data_pool; init_model=fitresult)
         report = (feature_importances=feature_importance(new_model),)