Package org.tribuo.multilabel.sgd.fm

package org.tribuo.multilabel.sgd.fm

Provides an implementation of a multi-label classification factorization machine model using Stochastic Gradient Descent.

Related Packages

Package	Description
org.tribuo.multilabel.sgd	Provides infrastructure for Stochastic Gradient Descent for multi-label classification problems.
org.tribuo.multilabel.sgd.linear	Provides an implementation of a multi-label classification linear model using Stochastic Gradient Descent.
org.tribuo.multilabel.sgd.objectives	Provides multi-label classification loss functions for Stochastic Gradient Descent.

Class	Description
FMMultiLabelModel	The inference time version of a multi-label factorization machine trained using SGD.
FMMultiLabelOptions	CLI options for training a linear classifier.
FMMultiLabelOptions.LossEnum	Available loss types.
FMMultiLabelTrainer	A trainer for a multi-label classification factorization machine using SGD.