Wybieranie istotnych cech
W tym ćwiczeniu wybierzesz tylko najważniejsze cechy, które zostaną użyte przez finalny model. Pamiętaj, że względne istotności cech są zapisane w kolumnie importance ramki danych relative_importances.
To ćwiczenie jest częścią kursu
HR Analytics: Przewidywanie rotacji pracowników w Pythonie
Instrukcje do ćwiczenia
- Wybierz tylko te cechy, których wartość
importancejest wyższa niż 1%. - Utwórz listę z tych cech i wyświetl ją (ten krok jest już za ciebie zrobiony).
- Korzystając z indeksu zapisanego w
selected_list, przekształćfeatures_trainifeatures_testtak, aby zawierały tylko cechy o istotności wyższej niż 1%.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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 = ____[____]