如何通过“信息增益”获得xgboost中的特征重要性?

2024-05-15 05:55:08 发布

您现在位置:Python中文网/ 问答频道 /正文

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

model.feature_importances_

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

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

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

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

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


Tags: 数据函数信息modelplottype增益特征

热门问题