Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix scikit learn cls doc. (#6041)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-08-21 10:23:06 +08:00 committed by GitHub
parent 1fd29edf66
commit 7be2e04bd4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
python-package/xgboost/sklearn.py
View File

@ -757,7 +757,10 @@ class XGBModel(XGBModelBase):
@xgboost_model_doc( @xgboost_model_doc(
"Implementation of the scikit-learn API for XGBoost classification.", "Implementation of the scikit-learn API for XGBoost classification.",
['model', 'objective']) ['model', 'objective'], extra_parameters='''
n_estimators : int
Number of boosting rounds.
''')
class XGBClassifier(XGBModel, XGBClassifierBase): class XGBClassifier(XGBModel, XGBClassifierBase):
# pylint: disable=missing-docstring,invalid-name,too-many-instance-attributes # pylint: disable=missing-docstring,invalid-name,too-many-instance-attributes
def __init__(self, objective="binary:logistic", **kwargs): def __init__(self, objective="binary:logistic", **kwargs):
@ -1041,7 +1044,10 @@ class XGBRegressor(XGBModel, XGBRegressorBase):
@xgboost_model_doc( @xgboost_model_doc(
"scikit-learn API for XGBoost random forest regression.", "scikit-learn API for XGBoost random forest regression.",
['model', 'objective']) ['model', 'objective'], extra_parameters='''
n_estimators : int
Number of trees in random forest to fit.
''')
class XGBRFRegressor(XGBRegressor): class XGBRFRegressor(XGBRegressor):
# pylint: disable=missing-docstring # pylint: disable=missing-docstring
def __init__(self, learning_rate=1, subsample=0.8, colsample_bynode=0.8, def __init__(self, learning_rate=1, subsample=0.8, colsample_bynode=0.8,