python决策树可视化速度慢？

Question

想画个决策树出来，但是我的电脑跑不动，不知道为什么，是代码有问题还是很吃性能。

from IPython.display import Image
from sklearn.tree import export_graphviz
import pydotplus
# 提取一个决策树
estimator = model.estimators_[5]
#features名称
features = list(train.columns)
dot_data = export_graphviz(estimator, out_file=None,
                         feature_names=features,
                         class_names=features,
                         filled=True, rounded=True,
                         special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
# 使用ipython的终端jupyter notebook显示。
Image(graph.create_png())

Answer 1

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 看下这篇博客，也许你就懂了，链接：python实现 构建决策树（模型）

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^