fitControl <- trainControl(method = "repeatedcv",
number = 10,
repeats = 10,
classProbs = TRUE,
summaryFunction = twoClassSummary)
log.model <- train(y ~., data = pts.df[,2:8],
method = "glm",
trControl = fitControl)
pred.log <- predict(log.model, newdata = testing[,2:8], type="prob")[,2]
tree.model <- train(y ~., data = pts.df[,2:8],
method = "rpart",
trControl = fitControl)
pred.tree <- predict(tree.model, newdata=testing[,2:8], type="prob")[,2]
rf.model <- train(y ~., data = pts.df[,2:8],
method = "rf",
trControl = fitControl)
pred.rf <- predict(rf.model, newdata=testing[,2:8], type="prob")[,2]