Plot a Single XGBoost Decision Tree

前端 未结 4 1985
慢半拍i
慢半拍i 2021-01-05 04:13

I am using method on https://machinelearningmastery.com/visualize-gradient-boosting-decision-trees-xgboost-python/ to plot a XGBoost Decision Tree

from numpy         


        
4条回答
  •  攒了一身酷
    2021-01-05 04:33

    To add to Serk's answer, you can also resize the figure before displaying it:

    # ...
    plot_tree(model)
    fig = plt.gcf()
    fig.set_size_inches(18.5, 10.5)
    plt.show()
    

提交回复
热议问题