当使用sklearn的feature_selection函数进行特征选择时,有没有办法跟踪实际的特征名称,而不是默认的“f1”、“f2”等?我有大量的特征,无法手动跟踪。显然,我可以编写代码来实现这一点,但我很好奇是否有简单的选项可以设置。
回答:
如果你有一个pandas数据框,你可以返回由函数选择的列名,只需使用get_support方法即可。
这里有一个从官方文档中稍作修改的快速示例。
当使用sklearn的feature_selection函数进行特征选择时,有没有办法跟踪实际的特征名称,而不是默认的“f1”、“f2”等?我有大量的特征,无法手动跟踪。显然,我可以编写代码来实现这一点,但我很好奇是否有简单的选项可以设置。
回答:
如果你有一个pandas数据框,你可以返回由函数选择的列名,只需使用get_support方法即可。
这里有一个从官方文档中稍作修改的快速示例。