Uses of Class
org.tribuo.common.sgd.AbstractSGDTrainer
Package
Description
Provides an implementation of a classification factorization machine using Stochastic Gradient Descent.
Provides an implementation of a classification linear model using Stochastic Gradient Descent.
Provides the base classes for models trained with stochastic gradient descent.
Provides an implementation of a multi-label classification factorization machine model using Stochastic Gradient Descent.
Provides an implementation of a multi-label classification linear model using Stochastic Gradient Descent.
Provides an implementation of factorization machines for regression using Stochastic Gradient Descent.
Provides an implementation of linear regression using Stochastic Gradient Descent.
-
Uses of AbstractSGDTrainer in org.tribuo.classification.sgd.fm
Modifier and TypeClassDescriptionclass
A trainer for a classification factorization machine using SGD. -
Uses of AbstractSGDTrainer in org.tribuo.classification.sgd.linear
Modifier and TypeClassDescriptionclass
A trainer for a linear classifier using SGD.class
A logistic regression trainer that uses a reasonable objective, optimiser, number of epochs and minibatch size. -
Uses of AbstractSGDTrainer in org.tribuo.common.sgd
Modifier and TypeClassDescriptionclass
AbstractFMTrainer<T extends Output<T>,
U, V extends AbstractFMModel<T>> A trainer for a quadratic factorization machine model which uses SGD.class
AbstractLinearSGDTrainer<T extends Output<T>,
U, V extends AbstractLinearSGDModel<T>> A trainer for a linear model which uses SGD. -
Uses of AbstractSGDTrainer in org.tribuo.multilabel.sgd.fm
Modifier and TypeClassDescriptionclass
A trainer for a multi-label classification factorization machine using SGD. -
Uses of AbstractSGDTrainer in org.tribuo.multilabel.sgd.linear
Modifier and TypeClassDescriptionclass
A trainer for a multi-label linear model which uses SGD. -
Uses of AbstractSGDTrainer in org.tribuo.regression.sgd.fm
Modifier and TypeClassDescriptionclass
A trainer for a regression factorization machine using SGD. -
Uses of AbstractSGDTrainer in org.tribuo.regression.sgd.linear
Modifier and TypeClassDescriptionclass
A trainer for a linear regression model which uses SGD.