From 768f3c2d1ce67561861e35fb6625d76d394236a4 Mon Sep 17 00:00:00 2001 From: Sameer Aftab <53132294+Sameer-aftab@users.noreply.github.com> Date: Mon, 20 Oct 2025 07:20:13 +0500 Subject: [PATCH] Add Decision Tree Visualizer for Iris dataset Trains a tree on custom data and visualizes it. --- Machine Learning/Decision Tree Visualizer.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 Machine Learning/Decision Tree Visualizer.py 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)