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

Answer 2

您还可以使用:

from xgboost import plot_importance
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = (14, 7)
plot_importance(booster=booster)

Answer 3

只添加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)放大或缩小图像