按照XGBoost实战完全总结(下)中的代码运行时,尽管能够正常运行,且出了正确的图,但是将其代码在pycharm自带的控制台运行时发现最后几行代码出现问题:
# 显示重要特征
plot_importance(model)
plt.show()
xgb.plot_tree(model, num_trees=10)
plt.show()
这样的代码在控制台运行时,会阻碍代码的运行,在运行至di第一个show之后,将出现的第一个图关闭,发现下面的代码不能正常运行,百度了许久没有发现原因,后查阅工具书,发现这么一段话:
也就是说show本身就是阻碍程序运行的一行代码,因此,我将代码优化为下面的形式:
# 显示重要特征
plot_importance(model)
plt.savefig("figure1.png",dpi=120)
xgb.plot_tree(model, num_trees=10)
plt.savefig("fi