Экспорт графика для тестовых данных, по которым вы делаете прогнозы - PullRequest
Новая
       25

Экспорт графика для тестовых данных, по которым вы делаете прогнозы

0 голосов
/ 17 сентября 2018

Все учебные пособия, которые я вижу по каким-то причинам, отлично экспортируют graph_viz в обучающие данные.Я могу заставить это работать, и вот мой код: 

from sklearn.tree import DecisionTreeClassifier
DT = DecisionTreeClassifier(criterion = 'gini', max_depth = 5, 
max_features='sqrt')
Decision_Tree = DT.fit(X_train, y)
Y_pred=Decision_Tree.predict(X_test)

from io import StringIO
dot_data = StringIO()
export_graphviz(Decision_Tree, out_file='tree.dot',  
filled=True, rounded=True,
impurity=False,feature_names=X_test.columns,class_names=['Average','Won- 
Booked'])
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
from subprocess import call
call(['dot', '-Tpng', 'tree.dot', '-o', 'tree.png', '-Gdpi=60'])

Это все работает и не дает мне никакого сообщения об ошибке, однако создает дерево решений для обучающих данных вместо тестовых данных.Я думаю, что мне нужно изменить то, что я положил в export_graphviz вместо моего текущего Decision_Tree, но я не уверен, что?

...