Jeux d'entraînement et de test : diviser les données

Analytique RH : prédire l’attrition des employés en R

Anurag Gupta

People Analytics Practitioner

Qu'est-ce qu'un modèle ?

Diviser les données avec 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, ]

Passons à la pratique !

Analytique RH : prédire l’attrition des employés en R