library(caTools) sample <- sample.split(data$y, SplitRatio = 0.7) train_data <- subset(data, sample == TRUE) test_data <- subset(data, sample == FALSE)