/scalaCART

Lightweight Yet Fast Scala implementation of CART

Primary LanguageScala

Classification and Regression Tree (CART)

Jianbo Ye

This is only a sample implementation of (relatively standard) CART in Scala, which includes

  • Splitting Rules (using axis aligned classifier)
  • Class Assignment
  • Pruning Trees (based on resubstitution estimates)

See more descriptions here. To run

$ sbt test:run
$ dot -Tpng graph.dot -o graph.png

Reference