Python, PyDot und DecisionTree

Ich versuche zu visualisieren, meine DecisionTree, aber immer die Fehler
Der code ist:

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")

Und der Fehler ist

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

Kann jemand mir erklären was ist das problem? Vielen Dank!

  • Sie zeigt uns den code? Ich sehe nicht die wenn-Anweisung, die das traceback wird hingewiesen. Andere als die, offensichtlich die Methode startswith() erwartet entweder einen string als input "string" oder ein Tupel von strings ("st", "st2", "st3"). Sie bestanden auf der falschen Datentyp in die startswith () - Methode aufrufen. Entweder Sie sind nicht mit codecs.BOM_UTF8 richtig, oder Sie haben zu werfen es in einen string --> str ("codecs".BOM_UTF8)
InformationsquelleAutor Polly | 2015-07-03
  1. 5

    Ich hatte exakt das gleiche problem, und verbrachte ein paar Stunden versucht, dies herauszufinden. Ich kann nicht garantieren, was ich hier arbeiten werde, für andere, aber es kann sein, lohnt sich ein Schuss.

    1. Ich habe versucht die Installation von offiziellen pydot - Pakete, aber ich habe Python 3 und Sie ging einfach nicht. Nach der Suche nach einem Hinweis in einem thread von einem der vielen websites, die ich abgekocht durch, ich landete der Installation diese Gabel-repository von pydot.
    2. Ich ging zu graphviz.org und installiert die software auf meinem Windows 7-Maschine. Wenn Sie nicht über Windows, suchen unter Ihre Download-Bereich für Ihr system.
    3. Nach erfolgreicher Installation, in der Umgebungsvariablen (Control Panel\All Control Panel Items\System\Advanced system settings > klicken Sie auf Environment Variables - Taste > unter System variables fand ich die variable path > klicken Sie auf Edit... > ich fügte ;C:\Program Files (x86)\Graphviz2.38\bin am Ende in der Variable value: Feld.
    4. Zu bestätigen, ich kann jetzt mit dot Befehle in der Befehlszeile (Windows-Befehlsprozessor), tippte ich dot -V welche zurückgegeben dot - graphviz version 2.38.0 (20140413.2041).

    In dem code unten, halten Sie im Verstand, ich bin das Lesen eines dataframe aus meiner Zwischenablage. Sie könnte es Lesen aus Datei oder whathaveyou.

    In IPython Notebook:

    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())

    Python, PyDot und DecisionTree

    Alternativ, wenn Sie nicht mit IPython, können Sie generieren Ihr eigenes Bild von der Kommandozeile wie lange Sie haben, graphviz installiert (Schritt 2 oben). Mit meinem Beispiel-code oben, verwenden Sie diese Zeile nach der Montage des Modells:

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

    öffnen Sie dann die Eingabeaufforderung, wo die treepic.dot - Datei und geben Sie diese Befehlszeile ein:

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

    Ein .png-Datei erstellt werden soll mit Ihre Entscheidung Baum.

    InformationsquelleAutor Jarad
  2. 9

    Im Falle der Verwendung von Python 3 verwenden, verwenden Sie einfach pydotplus statt pydot. Es wird auch eine soft-installation von pip.

    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")
    • kann jemand diese auf die sklearn Dokumentation?
    • Dies ist die beste raten - danke, +1 ich habe es mit Image(graph.create_png()) auf Jupyter statt zu schreiben, es in ein pdf und arbeitete Linie ein Charme
    • Sie können auch dot_data = tree.export_graphviz(clf, out_file=None)
    InformationsquelleAutor Juliano ENS
  3. 0

    Den line-in der Frage ist zu überprüfen, ob der stream/die Datei wird als UTF-8 kodiert

    Statt:

    if data.startswith(codecs.BOM_UTF8):

    verwenden:

    if codecs.BOM_UTF8 in data:

    Werden Sie wahrscheinlich mehr Erfolg haben...

    • Man sollte beachten, dass diese beiden Linien sind nicht ganz gleichwertig, wenn Daten starten muss, um mit ihm dann die zweite könnte man nicht arbeiten.
    • Er ist auf der Suche nach einen unicode in eine string-Methode. Wahrscheinlich nicht funktioniert. Obwohl Sie möglicherweise nicht entspricht, wird die Stückliste in der Regel am Anfang einer Datei und nicht irgendwo anders (es sei denn, Sie wirklich, mussed up your Datei) finden Sie unter en.wikipedia.org/wiki/Byte_order_mark
    • Ich denke das problem ist in meinen Daten-Datei, hat jemand weiß, wie soll es Aussehen? Ich habe eine csv-Datei, wobei der erste string enthält die Namen der Attribute in jeder Spalte, und weitere strings die numerische Daten enthalten. Also meine X und Y sind die numerischen Daten aus einer Datei, ich habe Sie machen "skiprows=1" beim öffnen meiner Datei
    • ohne zu sehen, die Datei, wir gehen alle zu erraten. Sie brauchen, um mehr details zu liefern, wenn Sie wollen mehr über Konstruktive Antworten. Meine Antwort oben wird wahrscheinlich zu viel mit Ihrem ursprünglichen Thema, aber in einem bestimmten Kontext.
    InformationsquelleAutor Incognos
Schreibe einen Kommentar