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

improved docstring for folds in cv function

Browse Source
This commit is contained in:
zyxue 2016-04-09 10:21:56 -07:00
parent c791894668
commit 79b35da308
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/training.py
View File

@ -370,7 +370,7 @@ def cv(params, dtrain, num_boost_round=10, nfold=3, stratified=False, folds=None
Number of folds in CV.
stratified : bool
Perform stratified sampling.
folds : KFold or StratifiedKFold
folds : a KFold or StratifiedKFold instance
Sklearn KFolds or StratifiedKFolds.
metrics : string or list of strings
Evaluation metrics to be watched in CV.