8-poissonTC.R

# Train
doMC::registerDoMC(10)
knnMod <- train(myRec, data = trainSet,
                method = "knn",
                tuneGrid = data.frame(k = seq(5, 25, by = 4)),
                trControl = ctrl)

enetMod <- train(myRec, data = trainSet,
                 method = "glmnet",
                 tuneGrid = expand.grid(alpha = seq(0, 1, length.out = 5),
                                        lambda = seq(.5, 2, length.out = 5)),
                 trControl = ctrl)

svmMod <- train(myRec, data = trainSet,
                method = "svmRadial",
                tuneLength = 8,
                trControl = ctrl)

rfMod <- train(myRec, data = trainSet,
               method = "ranger",
               tuneLength = 8,
               num.trees = 5000,
               trControl = ctrl)

xgbMod <- train(myRec, data = trainSet,
                method = "xgbTree",
                tuneLength = 8,
                trControl = ctrl)

cubMod <- train(myRec, data = trainSet,
                method = "cubist",
                tuneLength = 8,
                trControl = ctrl)

modelList <- list("KNN" = knnMod,
                  "ENET" = enetMod,
                  "SVM" = svmMod,
                  "RF" = rfMod,
                  "XGB" = xgbMod,
                  "CUB" = cubMod)