Skip to content

GitLab

Unverified Commit 9503d3f9 authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

[doc] remove excess fit arguments from docstrings (#3330)

parent 8ab35255
Show whitespace changes
Inline Side-by-side
Showing with 11 additions and 3 deletions
python-package/lightgbm/sklearn.py
View file @ 9503d3f9
......@@ -761,8 +761,12 @@ class LGBMRegressor(LGBMModel, _LGBMRegressorBase):
return self
_base_doc = LGBMModel.fit.__doc__
fit.__doc__ = (_base_doc[:_base_doc.find('eval_class_weight :')]
_base_doc = (_base_doc[:_base_doc.find('group :')]
+ _base_doc[_base_doc.find('eval_set :'):])
_base_doc = (_base_doc[:_base_doc.find('eval_class_weight :')]
+ _base_doc[_base_doc.find('eval_init_score :'):])
fit.__doc__ = (_base_doc[:_base_doc.find('eval_group :')]
+ _base_doc[_base_doc.find('eval_metric :'):])
class LGBMClassifier(LGBMModel, _LGBMClassifierBase):
......@@ -834,7 +838,11 @@ class LGBMClassifier(LGBMModel, _LGBMClassifierBase):
callbacks=callbacks, init_model=init_model)
return self
fit.__doc__ = LGBMModel.fit.__doc__
_base_doc = LGBMModel.fit.__doc__
_base_doc = (_base_doc[:_base_doc.find('group :')]
+ _base_doc[_base_doc.find('eval_set :'):])
fit.__doc__ = (_base_doc[:_base_doc.find('eval_group :')]
+ _base_doc[_base_doc.find('eval_metric :'):])
def predict(self, X, raw_score=False, start_iteration=0, num_iteration=None,
pred_leaf=False, pred_contrib=False, **kwargs):
......
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