编程语言
首页 > 编程语言> > python – 从分类器中检索训练功能名称列表

python – 从分类器中检索训练功能名称列表

作者:互联网

有没有办法检索用于训练分类器的特征名称列表,一旦用fit方法训练了?我想在申请看不见的数据之前获得这些信息.
用于训练的数据是pandas DataFrame,在我的例子中,分类器是RandomForestClassifier.

解决方法:

根据文档和以前的经验,无法获得至少在其中一个分割中考虑的特征列表.

您是否担心自己不想使用所有功能进行预测,只考虑实际用于培训的功能?在这种情况下,我建议在拟合后列出feature_importances_并消除看似不相关的功能.然后训练一个只有相关特征的新模型,并使用这些特征进行预测.

标签:random-forest,python,pandas,scikit-learn
来源: https://codeday.me/bug/20191008/1872658.html