Package org.tribuo.common.sgd
package org.tribuo.common.sgd
Provides the base classes for models trained with stochastic gradient descent.
-
ClassDescriptionAbstractLinearSGDModel<T extends Output<T>>AbstractLinearSGDTrainer<T extends Output<T>, U>A trainer for a linear model which uses SGD.AbstractSGDModel<T extends Output<T>>A nominal tuple used to capture the prediction and the number of active features used by the model.A trainer for a model which uses SGD.SGDObjective<T>An interface for a loss function that can produce the loss and gradient incurred by a single prediction.