ÎncepețiÎncepe gratuit

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

Vezi cursul

Instrucțiuni pentru exercițiu

  • Selectează doar caracteristicile cu o valoare importance mai 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ât features_train, cât și features_test astfel î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 = ____[____]
Editează și rulează codul