From f3b42cef6a687834399f1efce77599dab9a12007 Mon Sep 17 00:00:00 2001 From: secret_dude Date: Wed, 19 May 2021 05:04:00 +0200 Subject: [PATCH] tree graph --- decisiontree.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/decisiontree.py b/decisiontree.py index efa6805..0425d41 100644 --- a/decisiontree.py +++ b/decisiontree.py @@ -41,4 +41,14 @@ y = df['Decision'] dtree = DecisionTreeClassifier() dtree = dtree.fit(X, y) +data = tree.export_graphviz(dtree, out_file=None, feature_names=features) +graph = pydotplus.graph_from_dot_data(data) +graph.write_png('mydecisiontree.png') + +img = pltimg.imread('mydecisiontree.png') +imgplot = plt.imshow(img) +plt.show() + + + #print(dtree.predict([[0, 1, 0, 0, 0, 1]])) \ No newline at end of file