Package org.tribuo.common.sgd

package org.tribuo.common.sgd

Provides the base classes for models trained with stochastic gradient descent.

Class	Description
AbstractFMModel<T extends Output<T>>	A quadratic factorization machine model trained using SGD.
AbstractFMTrainer<T extends Output<T>,U,V extends AbstractFMModel<T>>	A trainer for a quadratic factorization machine model which uses SGD.
AbstractLinearSGDModel<T extends Output<T>>	A linear model trained using SGD.
AbstractLinearSGDTrainer<T extends Output<T>,U,V extends AbstractLinearSGDModel<T>>	A trainer for a linear model which uses SGD.
AbstractSGDModel<T extends Output<T>>	A model trained using SGD.
AbstractSGDModel.PredAndActive	A nominal tuple used to capture the prediction and the number of active features used by the model.
AbstractSGDTrainer<T extends Output<T>,U,V extends Model<T>,X extends FeedForwardParameters>	A trainer for a model which uses SGD.
FMParameters	A Parameters for factorization machines.
SGDObjective<T>	An interface for a loss function that can produce the loss and gradient incurred by a single prediction.