python - 在IPython Notebook中显示决策树

我的目标是在IPython笔记本上显示决策树。我的问题是，当我试图渲染它时，它会打开一个新窗口，而我希望它被显示为内联（如matplotlib绘图）。
下面是我使用的代码：

def show_tree(decisionTree, out_file, feature_names):
    out_file = 'viz_tree/' + out_file
    export_graphviz(decisionTree, out_file=out_file, feature_names=feature_names)
    dot = ''
    with open(out_file, 'r') as file:
        for line in file:
            dot += line
    dot = Source(dot)
    return dot

decisionTree.fit(inputs, outputs)
d = show_tree(decisionTree, 'tree.dot', col_names)
d.render(view=True)

我知道这样做是有可能的，因为这个example。
你知道我怎么做吗？

最佳答案

我最后做的是从源代码安装pydot library（pip安装对于python3来说似乎是坏的），然后使用这个函数：

import io
from scipy import misc

def show_tree(decisionTree, file_path):
    dotfile = io.StringIO()
    export_graphviz(decisionTree, out_file=dotfile)
    pydot.graph_from_dot_data(dotfile.getvalue()).write_png(file_path)
    i = misc.imread(file_path)
    plt.imshow(i)

# To use it
show_tree(decisionT, 'test.png')

这会将图像保存在由文件路径定义的文件中。然后简单地将其作为png来显示。
希望它能帮助你们中的一些人！