Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix document. [skip ci] (#6669)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-02-02 20:43:31 +08:00 committed by GitHub
parent 411592a347
commit 9d62b14591
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
doc/tutorials/dask.rst
View File

@ -410,6 +410,8 @@ Other customization
XGBoost dask interface accepts other advanced features found in single node Python XGBoost dask interface accepts other advanced features found in single node Python
interface, including callback functions, custom evaluation metric and objective: interface, including callback functions, custom evaluation metric and objective:
.. code-block:: python
def eval_error_metric(predt, dtrain: xgb.DMatrix): def eval_error_metric(predt, dtrain: xgb.DMatrix):
label = dtrain.get_label() label = dtrain.get_label()
r = np.zeros(predt.shape) r = np.zeros(predt.shape)

4
python-package/xgboost/dask.py
View File

@ -1313,8 +1313,8 @@ def inplace_predict( # pylint: disable=unused-argument
Returns Returns
------- -------
prediction : prediction :
When input data is ``dask.array.Array`` or ``DaskDMatrix``, the return value is an When input data is ``dask.array.Array``, the return value is an array, when input
array, when input data is ``dask.dataframe.DataFrame``, return value can be data is ``dask.dataframe.DataFrame``, return value can be
``dask.dataframe.Series``, ``dask.dataframe.DataFrame`` or ``dask.array.Array``, ``dask.dataframe.Series``, ``dask.dataframe.DataFrame`` or ``dask.array.Array``,
depending on the output shape. depending on the output shape.
''' '''