Chọn các đặc trưng quan trọng
Trong bài tập này, nhiệm vụ của bạn là chỉ chọn những đặc trưng quan trọng nhất để dùng cho mô hình cuối cùng. Hãy nhớ rằng độ quan trọng tương đối được lưu trong cột importance của DataFrame tên là relative_importances.
Bài tập này là một phần của khóa học
Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python
Hướng dẫn bài tập
- Chỉ chọn các đặc trưng có giá trị
importancelớn hơn 1%. - Tạo một danh sách từ các đặc trưng đó và in ra (đã được làm sẵn cho bạn).
- Dùng chỉ mục lưu trong
selected_list, biến đổi cảfeatures_trainvàfeatures_testđể chỉ bao gồm những đặc trưng có độ quan trọng lớn hơn 1%.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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 = ____[____]