选择重要特征
在本练习中,您的任务是仅选择将用于最终模型的最重要特征。请记住,相对重要性已保存在名为 relative_importances 的 DataFrame 的 importance 列中。
本练习是课程的一部分
HR Analytics:用 Python 预测员工流失
练习说明
- 仅选择
importance值高于 1% 的特征。 - 将这些特征创建为列表并打印(此步骤已为您完成)。
- 使用保存在
selected_list中的索引,转换features_train和features_test,使其仅包含重要性高于 1% 的特征。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# 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 = ____[____]