从经过训练的模型中提取特征名称

时间:2019-03-12 16:58:10

标签: python scikit-learn xgboost

我有一个从泡菜文件中读取的经过预先训练的XGBoost模型。当我尝试在模型的功能集之外的某些列上对新数据集进行预测时,我收到错误消息:

  

培训数据没有以下字段:column1,column2,...

我可以排除训练数据中不存在的这些列。与其对列名称(有很多)进行硬编码,我只是想找到训练列和预测数据集之间的交集。

有没有一种方法可以从训练后的模型中提取特征名称(显然该模型记录了字段名称)而无需返回训练数据集?

1 个答案:

答案 0 :(得分:1)

您可以按如下所示从腌制的模型中检索要素名称:

fitted_model.get_booster().feature_names