Update train_models.r

scripts/train_models.r

@@ -8,9 +8,6 @@ library("randomForest")
 # load in data
 load(file = "~/data.rdata")
 
-# remove RTN level
-data <- data %>% select(-c(RTN_LEVEL))
-
 # use index to partition data for training (90%) & validation
 training.index <- createDataPartition(data$RTN, p = 0.90, list = FALSE)