Question

我正在使用名为＆＃34; wine＆＃34;的数据集创建决策树：我正在尝试执行以下代码：

dt = c.fit(X_train, y_train)

创建决策树的图像：

def show_tree(tree, features, path):
    f = io.StringIO()
    export_graphviz(tree, out_file=f, feature_names=features)
    pydotplus.graph_from_dot_data(f.getvalue()).write_png(path)
    img = misc.imread(path)
    plt.rcParams["figure.figuresize"] = (20 , 20)
    plt.imshow(img)

调用图像：

show_tree(dt, features, 'dec_tree_01.png')

但是当我调用图像时会出现以下错误：

GraphViz's executables not found

我已从该网站安装了graphviz-2.38msi ...但同样的错误会不断显示。

我还在用户变量中添加了环境变量字符串，如下所示：

 %SystemRoot%\system32;%SystemRoot%;%SystemRoot%\System32\Wbem;%SYSTEMROOT%\System32\WindowsPowerShell\v1.0\;C:\Program Files (x86)\Graphviz2.38\bin;

但它也无法解决问题。

Answer 1

尝试在

这样的代码中追加os变量的路径

import os     
os.environ["PATH"] += os.pathsep + 'C:\Program Files (x86)\Graphviz2.38\bin'

注意：在执行show_tree（）

之前，请在代码顶部执行此操作

使用Python的决策树

1 个答案: