Uses of Class
org.tribuo.common.sgd.AbstractFMModel
Package
Description
Provides an implementation of a classification factorization machine 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 factorization machines for regression using Stochastic Gradient Descent.
-
Uses of AbstractFMModel in org.tribuo.classification.sgd.fm
Modifier and TypeClassDescriptionclass
The inference time version of a factorization machine trained using SGD. -
Uses of AbstractFMModel 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. -
Uses of AbstractFMModel in org.tribuo.multilabel.sgd.fm
Modifier and TypeClassDescriptionclass
The inference time version of a multi-label factorization machine trained using SGD. -
Uses of AbstractFMModel in org.tribuo.regression.sgd.fm
Modifier and TypeClassDescriptionclass
The inference time model of a regression factorization machine trained using SGD.