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

2019-06-21 12:44:42 +08:00
parent 77fc28427d
commit 30e1cb4e9e
1 changed files with 2 additions and 2 deletions
@@ -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