xgb获得特征重要性get_score()

model.get_score() 方法返回一个字典,其中包含模型中每个特征的得分。这些得分是通过计算特征在训练过程中用于划分数据的次数来计算的,因此可以用来衡量特征的重要性。model.get_score().keys() 是获取这个字典中的所有键(即特征名)。

在 XGBoost 中,每个特征都有一个"重要性得分",指的是该特征在决策树中被用作划分点的次数。model.get_fscore() 方法返回一个字典,其中包含每个特征的重要性得分。model.get_fscore().values() 是获取这个字典中的所有值(即每个特征的重要性得分),并将其用于构建一个列表

最后,将特征名和重要性得分组合成一个字典,其中键为特征名,值为该特征的重要性得分。

猜你喜欢

转载自blog.csdn.net/m0_62865498/article/details/130452749
今日推荐