tree visulization in decision tree(it is helpful for user)

[1998kajal]

(Note, you don't have to fill out every section here. They're just here for guidance. That said, nicely detailed feature requests are more likely to get eng attention sooner)

Problem

Is your feature request related to a problem? Please describe the problem here. Ex. I'm always frustrated when [...]

Solution

MVP: What's the smallest possible solution that would get 80% of the problem out of the way?

Possible additions: What are other things that could be added to the MVP over time to make it better?

Preferred solution: If you don't like the MVP above, tell us why, and what you'd like done instead.

Additional context

Add any other context or screenshots about the feature request here. For exmaple, did this FR come from https://discuss.streamlit.io or another site? Link the original source here!

[andfanilo]

Hello @1998kajal , thanks for the feature request !

Can you provide additional info on the library you're using to train Decision trees ? For example Scikit-learn Decision Tree provides bindings to Graphviz and Streamlit is able to render Graphviz objects so the following does the trick :

import graphviz as graphviz
from sklearn.datasets import load_iris
from sklearn import tree
import streamlit as st

X, y = load_iris(return_X_y=True)
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, y)

dot_data  = tree.export_graphviz(clf, out_file=None)

st.graphviz_chart(dot_data)

[randyzwitch]

Closing this issue, as Fanilo has highlighted that there is a solution and the issue has gone stale. If Fanilo's solution is not sufficient, I invite you to create an issue on our community forum and we can walk through the options there. Thanks for using Streamlit!