Skip to content

GitLab

Commit 13226fa3 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

add callbacks to LGBMRegressor

parent 72c2d790
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
python-package/lightgbm/sklearn.py
View file @ 13226fa3
...@@ -489,7 +489,7 @@ class LGBMRegressor(LGBMModel, LGBMRegressorBase): ...@@ -489,7 +489,7 @@ class LGBMRegressor(LGBMModel, LGBMRegressorBase):
eval_init_score=None, eval_init_score=None,
eval_metric="l2", eval_metric="l2",
early_stopping_rounds=None, verbose=True, early_stopping_rounds=None, verbose=True,
feature_name=None, categorical_feature=None): feature_name=None, categorical_feature=None, callbacks=None):
super(LGBMRegressor, self).fit(X, y, sample_weight=sample_weight, super(LGBMRegressor, self).fit(X, y, sample_weight=sample_weight,
init_score=init_score, eval_set=eval_set, init_score=init_score, eval_set=eval_set,
...@@ -498,7 +498,8 @@ class LGBMRegressor(LGBMModel, LGBMRegressorBase): ...@@ -498,7 +498,8 @@ class LGBMRegressor(LGBMModel, LGBMRegressorBase):
eval_metric=eval_metric, eval_metric=eval_metric,
early_stopping_rounds=early_stopping_rounds, early_stopping_rounds=early_stopping_rounds,
verbose=verbose, feature_name=feature_name, verbose=verbose, feature_name=feature_name,
categorical_feature=categorical_feature) categorical_feature=categorical_feature,
callbacks=callbacks)
return self return self
class LGBMClassifier(LGBMModel, LGBMClassifierBase): class LGBMClassifier(LGBMModel, LGBMClassifierBase):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment