Výběr důležitých příznaků
V tomto cvičení je tvým úkolem vybrat pouze nejdůležitější příznaky, které bude používat výsledný model. Nezapomeň, že relativní důležitosti jsou uloženy ve sloupci importance DataFrame s názvem relative_importances.
Toto cvičení je součástí kurzu
HR Analytics: Predicting Employee Churn in Python
Pokyny k cvičení
- Vyber pouze příznaky s hodnotou
importancevyšší než 1 %. - Vytvoř z těchto příznaků seznam a vypiš ho (tento krok je již hotový).
- Pomocí indexu uloženého v
selected_listupravfeatures_trainifeatures_testtak, aby obsahovaly pouze příznaky s důležitostí vyšší než 1 %.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# 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 = ____[____]