XGBoost for Python:绘制重要性

时间:2016-05-20 03:41:03

标签: python matplotlib xgboost

按照说明进行操作 https://xgboost.readthedocs.io/en/latest/python/python_intro.html

在构建我的模型后,我尝试使用Jupyter(使用Mac OS X,Chrome浏览器)

import matplotlib.pyplot as plt 
plt.style.use('ggplot') 
xgb.plot_importance(bst) 
xgb.plot_tree(bst, num_trees=2) 
xgb.to_graphviz(bst, num_trees=2)

但我有一些问题: to_graphviz确实给我一个情节,但它太大了,我看不出它。它在右侧切割

前两个命令(如果我在最后一行前放一个#)根本不显示任何情节。 是否有类似a.show()的东西?

相反,我得到以下输出

  

" matplotlib.axes._subplots.AxesSubplot at 0x12f0ee7d0" (在<>之间)

任何人都可以帮忙吗?谢谢

1 个答案:

答案 0 :(得分:2)

您似乎应该添加:plt.show()