xgboost，lightgbm权值图

最新推荐文章于 2023-08-25 00:50:39 发布

原创最新推荐文章于 2023-08-25 00:50:39 发布 · 444 阅读

0 ·

CC 4.0 BY-SA版权

计算机编程专栏收录该内容

17 篇文章

订阅专栏

本文介绍如何在使用Matplotlib绘制图表时，正确显示中文标签和标题，避免乱码问题。通过设置字体参数，实现lgb和xgb模型特征重要性图表的中文显示。

如果列名包含汉字，需要设置字体

import lightgbm as lgb
import xgboost as xgb
from matplotlib import pyplot as plt


####lgb，xgb训练完成后............

#设置字体
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus'] = False

# 使显示图标自适应
plt.rcParams['figure.autolayout'] = True
plt.figure(figsize=(16,10))

#lgb权值图
lgb.plot_importance(model_lgb, max_num_features=30)
plt.title("Featurertances")
plt.show()

#xgb权值图
from xgboost import plot_importance
plot_importance(model_xgb)