O'Reilly logo

Practical Predictive Analytics by Ralph Winters

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

Visual trees

Unless you are able to visualize the output, the decision tree rules can get difficult to interpret. Unless you implement the decision tree in the Scala language (which has a tree visualization method using the databricks Display command), some of your options are:

First parse the Spark output object to JSON format and then input using D3.js to visualize the tree. There are some prebuilt packages on Github, which can assist you in doing this.

Write the RDD you create to a file, input the RDD in Scala, and run DecisionTreeClassifier(). Then, fit the model and use the databricks display command on the model to display the tree.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required