Selectarea caracteristicilor importante
În acest exercițiu, sarcina ta este să selectezi doar cele mai importante caracteristici care vor fi folosite de modelul final. Reține că importanțele relative sunt salvate în coloana importance a DataFrame-ului numit relative_importances.
Acest exercițiu face parte din cursul
HR Analytics: Predicția fluctuației angajaților în Python
Instrucțiuni pentru exercițiu
- Selectează doar caracteristicile cu o valoare
importancemai mare de 1%. - Creează o listă din acele caracteristici și afișeaz-o (acest pas a fost deja realizat pentru tine).
- Folosind indexul salvat în
selected_list, transformă atâtfeatures_train, cât șifeatures_testastfel încât să includă doar caracteristicile cu o importanță mai mare de 1%.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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 = ____[____]