Préparer les ensembles train-test-validate
Dans cet exercice, vous allez mobiliser les outils vus jusqu’ici pour construire un modèle de classification visant à prédire l’attrition des employés.
Vous utiliserez le jeu de données attrition, qui contient 30 variables sur des employés et qui vous servira à prédire s’ils ont quitté l’entreprise.
Vous commencerez par préparer les jeux d’entraînement et de test, puis vous subdiviserez le jeu d’entraînement via une validation croisée afin de rechercher le meilleur modèle pour cette tâche.
Cet exercice fait partie du cours
<cours>Machine Learning dans le tidyverse</cours>Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
set.seed(42)
# Prepare the initial split object
data_split <- initial_split(___, prop = ___)
# Extract the training data frame
training_data <- training(___)
# Extract the testing data frame
testing_data <- testing(___)