从分类器中检索训练功能名称列表
问题内容:
一旦使用方法训练分类器,是否有办法检索用于分类器训练的特征名称列表fit
?我希望在申请看不见的数据之前先获得此信息。用于训练的数据是熊猫DataFrame
,在我的情况下,分类器是RandomForestClassifier
。
问题答案:
根据文档和先前的经验,无法获得至少在拆分中考虑的功能的列表。
您是否担心不想将所有功能都用于预测,而只是想将其实际用于训练?在这种情况下,我建议列出feature_importances_
之后的配件,并消除似乎不相关的功能。然后训练仅具有相关特征的新模型,并将这些特征也用于预测。