Package org.tribuo.classification.sgd.fm

package org.tribuo.classification.sgd.fm

Provides an implementation of a classification factorization machine using Stochastic Gradient Descent.

Related Packages

Package	Description
org.tribuo.classification.sgd	Provides infrastructure for Stochastic Gradient Descent for classification problems.
org.tribuo.classification.sgd.crf	Provides an implementation of a linear chain CRF trained using Stochastic Gradient Descent.
org.tribuo.classification.sgd.kernel	Provides a SGD implementation of a Kernel SVM using the Pegasos algorithm.
org.tribuo.classification.sgd.linear	Provides an implementation of a classification linear model using Stochastic Gradient Descent.
org.tribuo.classification.sgd.objectives	Provides classification loss functions for Stochastic Gradient Descent.

Class	Description
FMClassificationModel	The inference time version of a factorization machine trained using SGD.
FMClassificationOptions	CLI options for training a factorization machine classifier.
FMClassificationOptions.LossEnum	Available loss types.
FMClassificationTrainer	A trainer for a classification factorization machine using SGD.
TrainTest	Build and run a classifier for a standard dataset using FMClassificationTrainer.
TrainTest.TrainTestOptions	Command line options.