xgboost,lightgbm權值圖

如果列名包含漢字,需要設置字體

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)

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章