python - xgboost 图重要性图大小

Question

如何更改 xgboost 的绘图重要性函数的图形大小？

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

Answer 1

您可以在 plot_importance() 的 ax 参数中传递一个轴。例如，使用这个包装器:

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()