2016-10-17 8 views
3

xgboostのプロット重要度関数のFigureのサイズはどのように変更できますか?xgboostプロットの重要度の数値

不明な属性を除いて、figsize =(10,20)を渡そうとすると失敗します。

+0

を使用/chris/01_Analysis.ipynb –

答えて

8

引数の軸を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) 

はその後https://github.com/ViennaKaggle/allstate-claims-severity/blob/masterに基づいて、私は自分自身をロールアップ終了my_plot_importance()代わりのplot_importance()

関連する問題