如何通过'信息获取'在xgboost中获得功能重要性?

dsl*_*990 4 python machine-learning matplotlib python-3.x xgboost

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

model.feature_importances_
Run Code Online (Sandbox Code Playgroud)

但这不是我想要的.我希望通过信息获得重要性.

我们可以通过'获得'图来获得特征重要性:

xgboost.plot_importance(model, importance_type='gain')
Run Code Online (Sandbox Code Playgroud)

但是,我不知道如何从上面的图中获取特征重要性数据.要么

如果有功能,model.feature_importances_以获得增益功能的重要性?这两种方式中的任何一种都可行.如果问题不明确,请在评论中告诉我

dsl*_*990 7

你可以从中得到它

model.booster().get_score(importance_type='gain')
Run Code Online (Sandbox Code Playgroud)

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

  • model.get_booster()。get_score(importance_type ='gain')为我工作。。。看起来好像已经更新了。 (2认同)