Merge pull request #1098 from zyxue/patch-1

improved docstring for folds in cv function
This commit is contained in:
Yuan (Terry) Tang 2016-04-09 14:09:10 -04:00
commit 59610c49df

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.