Python, PyDot ve DecisionTree

Question

benim DecisionTree görselleştirmek için çalışıyorum, ama hata alıyorum kodudur:

X = [i[1:] for i in dataset]#attribute 
y = [i[0] for i in dataset] 
clf = tree.DecisionTreeClassifier() 

dot_data = StringIO() 
tree.export_graphviz(clf.fit(train_X, train_y), out_file=dot_data) 
graph = pydot.graph_from_dot_data(dot_data.getvalue()) 
graph.write_pdf("tree.pdf") 

Ve hata

Traceback (most recent call last): 
if data.startswith(codecs.BOM_UTF8): 
TypeError: startswith first arg must be str or a tuple of str, not bytes 

kimse bana açıklayabilir whats sorun? Çok teşekkür ederim!

Söz konusu hat akışı/dosya olarak kodlanmış olup olmadığını görmek için kontrol ediyor

Answer 1

Aynı problemi yaşadım ve sadece birkaç saat t onun dışarı. Burada paylaştığım şeyin başkaları için işe yarayacağını garanti edemiyorum ama bu bir çekim olabilir.

1. Ben resmi pydot paketlerin kurulumu denedim ama Python 3 var ve onlar sadece işe yaramadı. Aradan geçen birçok web sitesinden bir ipte bir not bulduktan sonra, this forked repository of pydot'u yükledim.
2. graphviz.org'a gittim ve yazılımlarını Windows 7 makinem üzerine yükledim. Windows yoksa, sisteminizin İndirme bölümünün altına bakın. Başarılı sonra
3. > Ortam değişkenleri (Control Panel\All Control Panel Items\System\Advanced system settings yılında, yüklemek System variables altında Environment Variables düğmesi> tıklayın ben değişken path bulundu>Edit... tıklayın> Ben Variable value: alanda sonuna ;C:\Program Files (x86)\Graphviz2.38\bin eklendi.
4. Şimdi dot komutları kullanabilirsiniz onaylamak için Komut Satırı (Windows Komut İşlemci) 'de, ben dot - graphviz version 2.38.0 (20140413.2041) döndü dot -V yazdınız.

aşağıdaki kodda, benim panodan bir dataframe okuyorum unutmayın. Olabilirsin dosyadan veya whathaveyou'dan okumak. IPython kullanmıyorsanız

import pandas as pd 
import numpy as np 
from sklearn import tree 
import pydot 
from IPython.display import Image 
from sklearn.externals.six import StringIO 

df = pd.read_clipboard() 
X = df[df.columns[:-1]] 
y = df[df.columns[-1]] 

dtr = tree.DecisionTreeRegressor(max_depth=3) 
dtr.fit(X, y) 

dot_data = StringIO() 
tree.export_graphviz(dtr, out_file=dot_data, feature_names=X.columns) 
graph = pydot.graph_from_dot_data(dot_data.getvalue()) 
Image(graph.create_png()) 

Alternatif olarak, (sürece graphviz kurduktan olarak komut satırından kendi resminizi oluşturabilir:

ipython Notebook yılında yukarıdaki adım 2). Yukarıda benim aynı örnek kodu kullanarak, modelin yerleştirilmesi sonra bu hattı kullanın:

tree.export_graphviz(dtr.tree_, out_file='treepic.dot', feature_names=X.columns) 

sonra treepic.dot dosyanın nerede komut istemini açmak ve bu komut satırını girin:

dot -T png treepic.dot -o treepic.png 

bir .png dosyası karar ağacınızla oluşturulmalıdır.

Answer 2

UTF-8

yerine:

if data.startswith(codecs.BOM_UTF8): 

kullanımı: Büyük olasılıkla olacak

if codecs.BOM_UTF8 in data: 

devamı ...

Answer 3

Python 3 kullanılması durumunda, sadece pydotplus yerine pydot kullanın. Ayrıca pip ile yumuşak bir yükleme işlemine sahip olacak.

import pydotplus 

<your code> 

dot_data = StringIO() 
tree.export_graphviz(clf, out_file=dot_data) 
graph = pydotplus.graph_from_dot_data(dot_data.getvalue()) 
graph.write_pdf("iris.pdf")