diff --git a/Machine Learning/Decision Tree Visualizer.py b/Machine Learning/Decision Tree Visualizer.py new file mode 100644 index 0000000..0c25c8e --- /dev/null +++ b/Machine Learning/Decision Tree Visualizer.py @@ -0,0 +1,15 @@ +from sklearn import tree +from sklearn.datasets import load_iris +import graphviz + +iris = load_iris() +clf = tree.DecisionTreeClassifier() +clf = clf.fit(iris.data, iris.target) + +dot_data = tree.export_graphviz(clf, out_file=None, + feature_names=iris.feature_names, + class_names=iris.target_names, + filled=True, rounded=True) + +graph = graphviz.Source(dot_data) +graph.render("decision_tree_visualization", view=True)