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

Fix cls typo. (#6247)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-10-16 16:40:44 +08:00 committed by GitHub
parent 3098d7cee0
commit 52452bebb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python-package/xgboost/sklearn.py
View File

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