How can decision tree model in Spark (pyspark) be visualized?

user_4780852 0
Viewed 3317

I am trying to visualize decision tree structure in pyspark. But all the tools are for data. I could not find any for visualizing tree structure. Or is there a way I can visualize using the rules from toDebugString?

dtreeviz pyspark apache-spark decision-tree python
2 Answers

I'm also a bigdata/ml engineer and I had many time the need to visualize the decision trees from Spark.

We've just released dtreeviz 1.1 , which includes support for Spark. You can now visualize the tree structure, leaf information, prediction path and more. Just check this notebook for more examples.

enter image description here enter image description here

user_45732500
Related