Advertisement
Guest User

Untitled

a guest
Jun 20th, 2019
78
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.59 KB | None | 0 0
  1. modellist_fit <- caretEnsemble::caretList(
  2. returnShipment ~.,
  3. data = trainSet,
  4. trControl = caret::trainControl(
  5. method = "boot",
  6. number = 2,
  7. savePredictions = TRUE,
  8. classProbs = TRUE,
  9. index = caret::createResample(trainSet$returnShipment, 2),
  10. verboseIter = TRUE
  11. ),
  12. tuneList = list(
  13. xgb = caretEnsemble::caretModelSpec(method = "xgbTree", tuneGrid = expand.grid(nrounds = 50, max_depth = 2, eta = 0.3, gamma = 0, colsample_bytree = 0.6, min_child_weight = 1, subsample = 1)),
  14. nnet = caretEnsemble::caretModelSpec(method = "nnet", tuneGrid = expand.grid(size = 15, decay = 0.1))
  15. )
  16. )
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement