Skip to content
Snippets Groups Projects
Commit f9e158a1 authored by innes's avatar innes
Browse files

Comment on and clean up plot_decision_tree.py

parent a398ad64
No related branches found
No related tags found
No related merge requests found
# this module plots a decision tree after training the classifier on the data
import re
import os
......@@ -31,15 +33,6 @@ tree.plot_tree(dtcf,
class_names=LANG_LIST,
filled=True)
plt.show()
"""
fig = plt.figure()
a = tree.plot_tree(dtcf,
class_names=LANG_LIST,
filled=True)
#fig.title("Decision tree")
fig.savefig(CWD+"/project/images/decision_tree.png")
fig.show()
"""
text_repr = tree.export_text(dtcf)
print(text_repr)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment