我们可以通过“权重”获取特征重要性:
model.feature_importances_
但这并不是我想要的。我想要通过信息增益来获取特征重要性。
我们可以通过“增益”图来获取特征重要性:
xgboost.plot_importance(model, importance_type='gain')
然而,我不知道如何从上面的图中获取特征重要性数据。或者
是否有类似于 model.feature_importances_
的函数可以提供基于增益的特征重要性?这两种方法都可以。 如果问题不够清楚,请在评论中告诉我。
回答:
你可以从以下方法获取:
model.booster().get_score(importance_type='gain')
http://xgboost.readthedocs.io/en/latest/python/python_api.html