想画个决策树出来,但是我的电脑跑不动,不知道为什么,是代码有问题还是很吃性能。
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())