如何更改xgboost的绘图重要性函数的数字大小?
尝试传递figsize =(10,20)失败,但未知属性除外。
答案 0 :(得分:15)
您可以在ax
中的plot_importance()
参数中传递轴。例如,使用此包装器:
def my_plot_importance(booster, figsize, **kwargs):
from matplotlib import pyplot as plt
from xgboost import plot_importance
fig, ax = plt.subplots(1,1,figsize=figsize)
return plot_importance(booster=booster, ax=ax, **kwargs)
然后使用my_plot_importance()
代替plot_importance()
答案 1 :(得分:0)
您还可以通过以下方式设置图形大小:
from xgboost import plot_importance
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = (14, 7)
plot_importance(booster=booster)
答案 2 :(得分:0)
仅将plt.rcParams["figure.figsize"] = (20,50)
添加到您的代码中
例如:
from xgboost import plot_importance
plt.figure(figsize=(40,20))
plot_importance(model,max_num_features=100)
plt.rcParams["figure.figsize"] = (20,100)
plt.show()
调整(20,100)可以放大或缩小图像大小