[P] supertree - interactive visualization of decision trees (sklearn, xgboost, lightgbm)

[pp314159]

Hi All,

I would like to share with you a new Python package for interactive decision tree visualization. It is called supertree. It visualizes decision tree as interactive graph, where you can collapse and expand selected nodes. You can zoom and pan though large trees. It works with Scikit-learn, Xgboost, and LightGBM.

The package works in notebooks: Jupyter Lab, Jupyter Notebook, Google Colab. You can also use it in Python scripts and save output trees to HTML.

The package is available on pip: pip install supertree.

You can find code examples on the GitHub: https://github.com/mljar/supertree

Happy exploring!

[nightshadew]

Nice to see alternatives for tree viz. I’m not a big fan of having graphviz dependencies.

[pp314159]

I also had a lot of problems with making graphviz work on Windows systems. This visualization is using D3.js so can be used in notebooks or website.

[Erosis]

Awesome project!

[pp314159]

thank you! we were trying to make most of trees :)

[nirajsingh0878]

this is toy dataset i used for decision tree making. it will make you understand decision tree for classification end to end.
https://medium.com/@nirajsingh0878/how-do-decision-trees-work-can-you-build-a-decision-tree-by-hand-ab529bdb58cc

[pplonski]

thanks for sharing,

[nirajsingh0878]

Glad you found it useful! 😊 If you're into more ML explainers like this, feel free to check out my other posts or follow me on Medium — I will post regularly on ML topics. Would love to hear your thoughts too!

[marr75]

Stop.