Training and testing datasets: splitting data

HR Analytics: Predicting Employee Churn in R

Anurag Gupta

People Analytics Practitioner

What is a model?

Splitting data with caret

# Load caret
library(caret)

# Set seed
set.seed(567)

# Store row numbers for training dataset
index_train <- createDataPartition(emp_final$turnover, p = 0.5, list = FALSE)

# Create training dataset
train_set <- emp_final[index_train, ]

# Create testing dataset
test_set <- emp_final[-index_train, ]

Let's practice!

HR Analytics: Predicting Employee Churn in R