Dela upp data i tränings- och testmängder
Det första steget när du tränar en modell är att dela upp data i en tränings- och en testmängd. Paketet tidymodels gör detta enkelt. Genom att avsätta en testmängd kan du utvärdera den tränade modellen på data som modellen aldrig har sett tidigare.
Du kommer att använda data om anställda inom hälso- och sjukvård, som innehåller information om huruvida de har lämnat företaget eller inte. Datamängden finns tillgänglig i attrition_df. Målvariabeln är Attrition.
Paketen tidyverse och tidymodels har redan laddats in.
Den här övningen är en del av kursen
Dimensionsreduktion i R
Övningsinstruktioner
- Initiera en uppdelning av data där 80 % används för träning och stratifiera baserat på
Attrition, målvariabeln. - Extrahera träningsdatamängden och spara den i
train. - Extrahera testdatamängden och spara den i
test.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Initialize the split
split <- ___(___, ___ = ___, strata = ___)
# Extract training set
train <- ___ %>% ___()
# Extract testing set
test <- ___ %>% ___()