Визуализация дерева решений в scikit-learn
Я пытаюсь спроектировать простое дерево решений, используя scikit-learn на Python (я использую Iacthon Notebook от Anaconda с Python 2.7.3 для ОС Windows) и визуализирую его следующим образом:
from pandas import read_csv, DataFrame
from sklearn import tree
from os import system
data = read_csv('D:/training.csv')
Y = data.Y
X = data.ix[:,"X0":"X33"]
dtree = tree.DecisionTreeClassifier(criterion = "entropy")
dtree = dtree.fit(X, Y)
dotfile = open("D:/dtree2.dot", 'w')
dotfile = tree.export_graphviz(dtree, out_file = dotfile, feature_names = X.columns)
dotfile.close()
system("dot -Tpng D:.dot -o D:/dtree2.png")
Однако я получаю следующую ошибку:
AttributeError: 'NoneType' object has no attribute 'close'
Я использую следующий блог в качестве ссылки:Ссылка на блог
Следующий вопрос stackoverflow, похоже, не работает и для меня:Вопрос
Может ли кто-нибудь помочь мне с тем, как визуализировать дерево решений в scikit-learn?