diff --git a/src/MLJCatBoostInterface.jl b/src/MLJCatBoostInterface.jl index 6170db8..919a9d8 100644 --- 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),)