Fix cls typo. (#6247)

2020-10-16 16:40:44 +08:00
parent 3098d7cee0
commit 52452bebb9
1 changed files with 3 additions and 3 deletions
--- 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).