Дерево решений Python не может визуализировать

Я использую Python 2.7 построить дерево решений, и я хочу визуализировать его, но всегда есть ошибка

IndexError: list index out of range

или же

OSError: [Errno 2] "dot.exe" not found in path.

происходят. Кто-нибудь знает как это решить? Спасибо!!!

Вот мой код:

from sklearn import tree
from sklearn.externals.six import StringIO
from sklearn.tree import export_graphviz
from sklearn.model_selection import train_test_split
from sklearn.metrics import classification_report, confusion_matrix
from sklearn.tree import DecisionTreeClassifier
import pydot
import pandas as pd  
import numpy as np  
import matplotlib.pyplot as plt
import random
from numpy.random import RandomState
from scipy import stats
import math
import graphviz
dataset =pd.read_csv(r"C:\Users\US1\Desktop\as21\skysurvey\skysurvey\training_data.csv")
dataset.head() 
X = dataset.drop('class', axis=1)  
y = dataset['class']
z=['ra', 'dec', 'u', 'g', 'r', 'i', 'z', 'run', 'rerun', 'camcol', 'field', 'specobjid', 'redshift', 'plate', 'mjd', 'fiberid']
a= ['STAR' 'GALAXY' 'QSO']
classifier = tree.DecisionTreeClassifier(criterion='gini')  
classifier = classifier.fit(X, y)
dot_data = StringIO()
tree.export_graphviz(classifier, out_file=dot_data, 
                 feature_names = z,  
                 class_names = a,  
                 filled=True, rounded = True,  
                 special_characters = True  
                )
graph = pydot.graph_from_dot_data(dot_data.getvalue())
graph[0].write_pdf("DecisionTree.pdf")

0 ответов

Другие вопросы по тегам