Package org.tribuo.regression.sgd.fm

package org.tribuo.regression.sgd.fm

Provides an implementation of factorization machines for regression using Stochastic Gradient Descent.

Related Packages

Package	Description
org.tribuo.regression.sgd	Provides infrastructure for Stochastic Gradient Descent based regression models.
org.tribuo.regression.sgd.linear	Provides an implementation of linear regression using Stochastic Gradient Descent.
org.tribuo.regression.sgd.objectives	Provides regression loss functions for Stochastic Gradient Descent.

Class	Description
FMRegressionModel	The inference time model of a regression factorization machine trained using SGD.
FMRegressionTrainer	A trainer for a regression factorization machine using SGD.
TrainTest	Build and run a regression factorization machine for a standard dataset.
TrainTest.FMRegressionOptions	Command line options.
TrainTest.LossEnum	Loss function.