Description Usage Arguments Examples
Train a stacked model using caret
1 | trainStack(x, y, layers, folds, verbose = F)
|
x |
Predictors |
y |
Response |
layers |
Stack layers. A list of lists of models. |
folds |
CV folds, as created by caret::createFolds |
verbose |
Output progress messages |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | stackLayers <- list(
list(
gbm2 = list(
parallel = 4,
params = list(
method = "gbm",
tuneGrid = expand.grid(
n.trees = 300,
interaction.depth = 2,
shrinkage = 0.1,
n.minobsinnode = 10
),
trControl = trainControl(method = "none")
)
),
gbm10 = list(
parallel = 4,
params = list(
method = "gbm",
tuneGrid = expand.grid(
n.trees = 300,
interaction.depth = 10,
shrinkage = 0.1,
n.minobsinnode = 10
),
trControl = trainControl(method = "none")
)
)
),
list(
gbm2 = list(
parallel = 4,
params = list(
method = "gbm",
tuneGrid = expand.grid(
n.trees = 300,
interaction.depth = 2,
shrinkage = 0.1,
n.minobsinnode = 10
),
trControl = trainControl(method = "none")
)
)
)
)
folds <- caret::createFolds(x, 10)
stackModel <- trainStack(x, y, stackLayers, folds)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.