小编Gie*_*itė的帖子

XGBoost 中的绘图编号格式plot_importance()

我训练了一个 XGBoost 模型,并使用plot_importance() 来绘制训练模型中最重要的特征。尽管如此,图中的数字有几个小数值,这些值淹没了绘图并且不适合绘图。

我已经搜索了绘图格式选项,但我只找到了如何格式化轴(尝试格式化 X 轴,希望它也能格式化相应的轴)

我在 Jupyter Notebook 中工作(如果这有什么区别的话)。代码如下:

xg_reg = xgb.XGBClassifier(
                objective = 'binary:logistic',
                colsample_bytree = 0.4,
                learning_rate = 0.01,
                max_depth = 15, 
                alpha = 0.1, 
                n_estimators = 5,
                subsample = 0.5,
                scale_pos_weight = 4
                )
xg_reg.fit(X_train, y_train) 
preds = xg_reg.predict(X_test)

ax = xgb.plot_importance(xg_reg, max_num_features=3, importance_type='gain', show_values=True) 

fig = ax.figure
fig.set_size_inches(10, 3)
Run Code Online (Sandbox Code Playgroud)

我有什么遗漏的吗?是否有任何格式化函数或参数要传递?

我希望能够格式化特征重要性分数,或者至少删除小数部分(例如“25”而不是“25.66521”)。下面附上当前的图。

xgboost_feature_importance_scores

python plot matplotlib boosting xgboost

3
推荐指数
1
解决办法
9489
查看次数

标签 统计

boosting ×1

matplotlib ×1

plot ×1

python ×1

xgboost ×1