diff --git a/optgbm/sklearn.py b/optgbm/sklearn.py index 0eb9aa0..4a2f6b3 100644 --- a/optgbm/sklearn.py +++ b/optgbm/sklearn.py @@ -638,6 +638,7 @@ def fit( feature_name=feature_name, categorical_feature=categorical_feature, callbacks=callbacks, + init_model=init_model, ) elapsed_time = time.perf_counter() - start_time