无法在Jupyter笔记本中呈现GraphViz图像

时间:2019-03-07 06:38:56

标签: python jupyter-notebook graphviz

我正在尝试渲染使用sklearn的{​​{1}}创建的决策树对象。我正在使用这样的东西生成这棵树:

DecisionTreeClassifier

其中tree = DecisionTreeClassifier(max_depth=3, random_state=17).fit(X_train, y_train)X_train分别是2D和1D数据帧。然后,我尝试在y_train中渲染树,如下所示:

jupyter-notebook

我得到的不是输出渲染的图像,而是输出以下文本:

from sklearn.tree import export_graphviz
from io import StringIO
import pydotplus
from ipywidgets import Image

dot_data = StringIO()
export_graphviz(decision_tree=tree, feature_names=df.drop('Churn', axis=1).columns, 
                out_file=dot_data, filled=True)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
Image(value=graph.create_png())

为什么我得到这个而不是实际的图像?在this video的1:04:22似乎很好...

0 个答案:

没有答案
相关问题