From 79b35da3082ee722f37723f3e71549889f77953f Mon Sep 17 00:00:00 2001 From: zyxue Date: Sat, 9 Apr 2016 10:21:56 -0700 Subject: [PATCH] improved docstring for folds in cv function --- python-package/xgboost/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/xgboost/training.py b/python-package/xgboost/training.py index d84c030f9..929a79f22 100644 --- a/python-package/xgboost/training.py +++ b/python-package/xgboost/training.py @@ -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.