Välja viktiga särdrag
I den här övningen är din uppgift att välja ut enbart de viktigaste särdragen som den slutliga modellen ska använda. Kom ihåg att de relativa viktningarna är sparade i kolumnen importance i DataFrame:n relative_importances.
Den här övningen är en del av kursen
HR-analys: Förutsäg personalomsättning i Python
Övningsinstruktioner
- Välj enbart de särdrag vars
importance-värde är högre än 1 %. - Skapa en lista av dessa särdrag och skriv ut dem (detta är redan gjort åt dig).
- Använd indexet sparat i
selected_listför att transformera bådefeatures_trainochfeatures_testså att de endast innehåller särdrag med en viktning högre än 1 %.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# select only features with relative importance higher than 1%
selected_features = relative_importances[relative_importances.____>0.01]
# create a list from those features: done
selected_list = selected_features.index
# transform both features_train and features_test components to include only selected features
features_train_selected = features_train[selected_list]
features_test_selected = ____[____]