2016-08-25 6 views
0

私は単純な例を再現できません。ここでは、それがどのようになるです:私は木をグラフ化しようとするとgraphvizでツリーをグラフ化する方法は?

import pandas as pd 
import numpy as np 
import sklearn as skl 
from sklearn import tree 
from sklearn.cross_validation import train_test_split as tts 



# import data and give a little overview 
sample = pd.read_stata('sample_data.dta') 
s = sample 

# Let's learn 
X = np.array((s.crisis, s.cash, s.industry, s.current_debt, s.activity)).reshape(1000, 5) 
y = np.array(s.wc_measure) 

X_train, X_test, y_train, y_test = tts(X, y, test_size = .8) 

my_tree = tree.DecisionTreeClassifier() 
clf = my_tree.fit(X_train, y_train) 
predictions = my_tree.predict(X_test) 

# Graph the tree 
from sklearn.externals.six import StringIO 
import pydotplus 
dotfile = StringIO() 
tree.export_graphviz(clf, out_file=dotfile) 
pydotplus.graph_from_dot_data(dotfile.getvalue()).write_png("my_tree.png") 

、私は次のエラーを取得する:

pydotplus.graphviz.InvocationException: GraphViz's executables not found 

問題が何であるか任意のアイデア?私はgraphvizのドキュメントにきちんと従っています。

+0

あなたの問題は単に 'tree'または任意のgraphvizグラフですか? –

+0

それはtree.exportまで実行されます:/ – Rachel

答えて

1

graphvizソフトウェアをインストールしていないようです。モジュールを使用する前にdownloadをインストールしてください。

+0

私は 'pip install graphviz'を使用しました...そうすべきでしょうか? – Rachel

+0

いいえ、それでは不十分です。 'pip install graphviz'は、graphvizソフトウェアを使用するpythonモジュールだけをインストールします。インストールセクションの[here](https://pypi.python.org/pypi/graphviz)を見てください。 –

関連する問題