Machine Learning with R Cookbook - Second Edition

How to do it...

Perform the following steps to generate an ROC curve of each fitted model:

1. Install and load the library, pROC:

        > install.packages("pROC")
        > library("pROC")  

2. Set up the training control with a 10-fold cross-validation in three repetitions:

        > control = trainControl(method = "repeatedcv",
        +                         number = 10,
        +                         repeats = 3,
        +                         classProbs = TRUE,
        +                         summaryFunction = twoClassSummary)  

3. Then, you can train a classifier on the training dataset using glm:

        > glm.model= train(churn ~ .,
        +                     data = trainset,
        +                     method = "glm",
        +                     metric = "ROC",
        +                     trControl = control)  

4. Also, you can train a classifier on the training dataset using svm:

 > svm.model= train(churn ~ ., + data = trainset, + method = "svmRadial", + metric = "ROC", + trControl ...