xgboost情节重要性数字大小

时间:2016-10-17 08:40:20

标签: python matplotlib scikit-learn xgboost

如何更改xgboost的绘图重要性函数的数字大小?

尝试传递figsize =(10,20)失败,但未知属性除外。

3 个答案:

答案 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)可以放大或缩小图像大小