Skip to content

Visualizes the Random Forest debug string from the MLLib in Spark using D3.js

Notifications You must be signed in to change notification settings

unnati-xyz/EurekaTrees

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EurekaTrees

This program was made to visualize MLLib Random Forests. Running this program with a supplied trees file generates visual version of the trees. You can get this from printing out the debug string of the generated model. In python that is: model.toDebugString()

Run:
python eurekatrees.py --trees ./sample_files/trees.txt

If you have a csv with the names of your features you can run that command with the columns switch.
python eurekatrees.py --trees ./sample_files/trees.txt --columns ./sample_files/columns.csv

The output are a bunch of HTML files. If you open up the file home.html it should allow you to easily navigate to each of your trees.

Example Tree: Example Tree

About

Visualizes the Random Forest debug string from the MLLib in Spark using D3.js

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • HTML 69.4%
  • Python 30.6%