python - 如何通过 'information gain' 获得 xgboost 中的特征重要性？

Question

我们可以通过“权重”获得特征重要性:

model.feature_importances_

但这不是我想要的。我想通过信息获取的重要性。

我们可以通过“增益”图获得特征重要性:

xgboost.plot_importance(model, importance_type='gain')

但是，我不知道如何从上面的图中获取特征重要性数据。或者

是否有像 model.feature_importances_ 这样的函数来赋予增益特征重要性？这两种方法中的任何一种都可以。如果问题不清楚，请在评论中告诉我

Answer 1

你可以从

model.booster().get_score(importance_type='gain')

http://xgboost.readthedocs.io/en/latest/python/python_api.html