私は単純な例を再現できません。ここでは、それがどのようになるです:私は木をグラフ化しようとすると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のドキュメントにきちんと従っています。
あなたの問題は単に 'tree'または任意のgraphvizグラフですか? –
それはtree.exportまで実行されます:/ – Rachel