Fix docstring for XGBModel.predict() [skip ci] (#4592)

2019-06-20 21:44:42 -07:00 · 2019-06-20 21:44:42 -07:00 · 30e1cb4e9e
commit 30e1cb4e9e
parent 77fc28427d
1 changed files with 2 additions and 2 deletions
--- a/python-package/xgboost/sklearn.py
+++ b/python-package/xgboost/sklearn.py
@ -431,8 +431,8 @@ class XGBModel(XGBModelBase):

        Parameters
        ----------
-        data : DMatrix
-            The dmatrix storing the input.
+        data : numpy.array/scipy.sparse
+            Data to predict with
        output_margin : bool
            Whether to output the raw untransformed margin value.
        ntree_limit : int