Package org.tribuo.multilabel.sgd.objectives

package org.tribuo.multilabel.sgd.objectives

Provides multi-label classification loss functions for 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.fm	Provides an implementation of a multi-label classification factorization machine model using Stochastic Gradient Descent.
org.tribuo.multilabel.sgd.linear	Provides an implementation of a multi-label classification linear model using Stochastic Gradient Descent.

Classes

Class	Description
BinaryCrossEntropy	A multilabel version of binary cross entropy loss which expects logits.
Hinge	Hinge loss, scores the correct value margin and any incorrect predictions -margin.