Визуализация дерева решений

Question

Я пытаюсь построить дерево решений в python и визуализировать его. Однако Graphviz создает проблему.

Вот код, который я попробовал:

# -*- coding: utf-8 -*- 
import pandas as pd 
from pandas import Series, DataFrame 
import numpy as np 
import os 
import matplotlib.pylab as plt 
import sklearn 
from sklearn.cross_validation import train_test_split 
from sklearn.tree import DecisionTreeClassifier 
import sklearn.metrics 
from sklearn.metrics import classification_report 
from sklearn import tree 
from io import BytesIO as StringIO 
from IPython.display import Image 
import pydot 
import pydotplus 
import graphviz 

path = "C:\\Users\\admin\\PycharmProjects\\conference\\New\\decision_tree\\" 

data = pd.read_csv(path + "weights.csv") 
data_clean = data.dropna() 

#data_clean.dtypes() 
#data_clean.describe() 

predictors = data_clean[['anxiety', 'taking']] 
target = data_clean['influence_score'] 

pred_train, pred_test, tar_train, tar_test = train_test_split(predictors,target, test_size=.2) 

#pred_train.shape 
#pred_test.shape 
#tar_test.shape 
#tar_train.shape 

classifier = DecisionTreeClassifier() 
classifier = classifier.fit(pred_train,tar_train) 

predictons = classifier.predict(pred_test) 

sklearn.metrics.confusion_matrix(tar_test,predictons) 

data = StringIO() 

#out_dot = path + "tree.dot" 
tree.export_graphviz(classifier, out_file = data) 

graph = pydotplus.graph_from_dot_data(data.getvalue()) 

out_tree = path + "tree_full.pdf" 

graph.write_pdf(out_tree) 

Однако я получаю эту ошибку.

File "C:\Python27\lib\site-packages\pydotplus\graphviz.py", line 1960, in create 
    'GraphViz\'s executables not found') 
pydotplus.graphviz.InvocationException: GraphViz's executables not found 

Я следовал всю процедуру установки Graphviz и pydotplus, включая изменение переменных окружения, а также импортировать их в питона IDE (я использую PyCharm). Версия Graphviz установлена ​​2.38.

Как это сделать.

Answer 1

Я решил. Кажется, я ошибался в обновлении переменных среды. работает отлично.