0

ディシジョン ツリーのビジュアル イメージを生成する際に問題が発生しています。私はpython 2.7を使用しており、graphvizとpydotをインストールしましたが、コードを実行すると、「視覚化を生成するためにdot、つまりgraphvizを実行できませんでした」というエラーが表示されます。何が問題なのですか?

NBはLinuxではなくWindowsを使用しています

以下は私のコードです

from __future__ import print_function
import pandas as pd
import os
import subprocess
from sklearn.tree import DecisionTreeClassifier, export_graphviz
os.chdir('C:\\Users\\lussi\\Desktop\\tests')
df= pd.read_csv('yield2.csv')
print("* df.head()", df.head(), sep="\n", end="\n\n")
features = list(df.columns[:6])
y = df["yield"]
X = df[features]
dt = DecisionTreeClassifier(min_samples_split=20, random_state=99)
dt.fit(X, y)
def visualize_tree(tree, f_names):

with open("dt.dot", 'w') as f:
    export_graphviz(tree, out_file=f,
                    feature_names=f_names)

command = ["dot", "-Tpng", "dt.dot", "-o", "dt.png"]
try:
    subprocess.check_call(command)
except:
    exit("Could not run dot, ie graphviz, to "
         "produce visualization")
visualize_tree(dt, features) 
4

0 に答える 0