开始使用免费开始使用

选择重要特征

在本练习中,您的任务是仅选择将用于最终模型的最重要特征。请记住,相对重要性已保存在名为 relative_importances 的 DataFrame 的 importance 列中。

本练习是课程的一部分

HR Analytics:用 Python 预测员工流失

查看课程

练习说明

  • 仅选择 importance 值高于 1% 的特征。
  • 将这些特征创建为列表并打印(此步骤已为您完成)。
  • 使用保存在 selected_list 中的索引,转换 features_trainfeatures_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 = ____[____]
编辑并运行代码