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).