Add functionality to decision trees in preparation for adding Random Forests

rinuboney / clatern

Machine Learning in Clojure

Eclipse Public License 1.0

67 stars 12 forks source link

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

rnowling commented 9 years ago

Fixed a few things I missed. Closing in favor of #36