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

make it possible to use a list of pre-defined CV folds in xgb.cv

Browse Source
This commit is contained in:
Vadim Khotilovich
2015-04-03 13:24:04 -05:00
parent c03b42054f
commit 31b0e53cd4
3 changed files with 42 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
R-package/man/xgb.cv.Rd
View File

@@ -6,7 +6,8 @@
\usage{
xgb.cv(params = list(), data, nrounds, nfold, label = NULL,
missing = NULL, prediction = FALSE, showsd = TRUE, metrics = list(),
obj = NULL, feval = NULL, stratified = TRUE, verbose = T, ...)
obj = NULL, feval = NULL, stratified = TRUE, folds = NULL,
verbose = T, ...)
}
\arguments{
\item{params}{the list of parameters. Commonly used ones are:
@@ -57,7 +58,10 @@ gradient with given prediction and dtrain.}
\code{list(metric='metric-name', value='metric-value')} with given
prediction and dtrain.}
\item{stratified}{\code{boolean}, whether sampling of folds should be stratified by the values of labels in \code{data}}
\item{stratified}{\code{boolean} whether sampling of folds should be stratified by the values of labels in \code{data}}
\item{folds}{\code{list} provides a possibility of using a list of pre-defined CV folds (each element must be a vector of fold's indices).
If folds are supplied, the nfold and stratified parameters would be ignored.}
\item{verbose}{\code{boolean}, print the statistics during the process}