Skip to content

Latest commit

 

History

History
13 lines (12 loc) · 483 Bytes

README.md

File metadata and controls

13 lines (12 loc) · 483 Bytes

decision_tree

A simple implementation of a decision tree made as practice while trying to wrap my head around Scala and functional programmming.

Future Work

  • Add support for splitting on continuous variables
  • Refactor so that splits are made on functions not specific values
  • Add Gini impurity
  • Add tests for fit and predict
    • Simple tests
  • Compare to performance of other packages
  • Extend for regression
  • Extend for random forests
  • Extend for gbm