set.seed(42) sample_split <- sample.split(Y = iris$Species, SplitRatio = 0.75) train_set <- subset(x = iris, sample_split == TRUE) test_set <- subset(x = iris, sample_split == FALSE)