Add functionality to decision trees in preparation for adding Random Forests - Githubissues

rinuboney / clatern

Machine Learning in Clojure

Eclipse Public License 1.0

67 stars 12 forks source link

Add functionality to decision trees in preparation for adding Random Forests #37

Closed rnowling closed 9 years ago

rnowling commented 9 years ago

Changes include:

Add optional parameter for function for choosing features to calculate splits on
Add optional parameters for list of element indices to use in training
Add ClassProbabilityEstimator protocol with predict-prob and predict-log-prob functions
Extend documentation with more algorithm details
Refactor internal predict functions
Rename decision-tree function to cart
Initialize depth to 0 instead of 1 so that max-depth matches expected semantics

Includes changes from discussion of #36.