From 7be2e04bd43ec1ed835e8fe0e3db476c017da7a9 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Fri, 21 Aug 2020 10:23:06 +0800 Subject: [PATCH] Fix scikit learn cls doc. (#6041) --- python-package/xgboost/sklearn.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index c6c34dce1..4721533b4 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -757,7 +757,10 @@ class XGBModel(XGBModelBase): @xgboost_model_doc( "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): # pylint: disable=missing-docstring,invalid-name,too-many-instance-attributes def __init__(self, objective="binary:logistic", **kwargs): @@ -1041,7 +1044,10 @@ class XGBRegressor(XGBModel, XGBRegressorBase): @xgboost_model_doc( "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): # pylint: disable=missing-docstring def __init__(self, learning_rate=1, subsample=0.8, colsample_bynode=0.8,