-
Notifications
You must be signed in to change notification settings - Fork 12
Description
Clatern's current interface has functions for training various types of models. The models are returned as records which implement the IFn
protocol so they can be called as functions to predict the classes of input vectors (and/or matrices, in some cases).
In addition to predicting a class label, some models can predict the probabilities that an input sample belongs to each class. In fact, this functionality is needed by Random Forests to make class label predictions from Decision Trees. As such, I created a new protocol ClassProbabilityEstimator
with predict-prob
and predict-log-prob
functions in PR #37.
After discussions on PR #37, Rinu and I decided to follow the example of core.matrix
and provide top-level predict
, predict-prob
, and predict-log-prob
functions that employ the protocols.