There is such a code, I get an error when starting
GraphViz \ 's executables not found
On the laptop (MacOS) Anaconda is already installed, I understand that the problem is that GraphViz not installed.
When I type in the terminal:
conda install -c anaconda graphviz=2.38.0 I get the error:
invalid package specification: graphviz = 2.38.
How to solve this problem?
import numpy as np from sklearn.datasets import load_iris from sklearn import tree iris = load_iris() test_idx =[0, 50, 100] #training data train_target =np.delete(iris.target, test_idx) train_data =np.delete(iris.data, test_idx, axis=0) #testing data test_target = iris.target[test_idx] test_data = iris.data[test_idx] clf = tree.DecisionTreeClassifier() clf.fit(train_data, train_target) print test_target print clf.predict(test_data) #viz code from sklearn.externals.six import StringIO import pydot dot_data = StringIO() tree.export_graphviz(clf,out_file=dot_data,feature_names=iris.feature_names, class_names=iris.target_names,filled=True, rounded=True, impurity=False) graph =pydot.graph_from_dot_data(dot_data.getvalue()) graph.write_pdf("iris.pdf")