我有一个从泡菜文件中读取的经过预先训练的XGBoost模型。当我尝试在模型的功能集之外的某些列上对新数据集进行预测时,我收到错误消息:
培训数据没有以下字段:column1,column2,...
我可以排除训练数据中不存在的这些列。与其对列名称(有很多)进行硬编码,我只是想找到训练列和预测数据集之间的交集。
有没有一种方法可以从训练后的模型中提取特征名称(显然该模型记录了字段名称)而无需返回训练数据集?
答案 0 :(得分:1)
您可以按如下所示从腌制的模型中检索要素名称:
fitted_model.get_booster().feature_names