From 52452bebb943e246f11465a79775ca43a430ba2e Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Fri, 16 Oct 2020 16:40:44 +0800 Subject: [PATCH] Fix cls typo. (#6247) --- python-package/xgboost/sklearn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index 96d358128..34e4d3fbd 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -879,7 +879,7 @@ class XGBClassifier(XGBModel, XGBClassifierBase): Parameters ---------- data : array_like - The dmatrix storing the input. + Feature matrix. output_margin : bool Whether to output the raw untransformed margin value. ntree_limit : int @@ -933,8 +933,8 @@ class XGBClassifier(XGBModel, XGBClassifierBase): Parameters ---------- - data : DMatrix - The dmatrix storing the input. + data : array_like + Feature matrix. ntree_limit : int Limit number of trees in the prediction; defaults to best_ntree_limit if defined (i.e. it has been trained with early stopping), otherwise 0 (use all trees).