org.tribuo.regression.sgd.objectives

Class Huber

java.lang.Object

org.tribuo.regression.sgd.objectives.Huber

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance>, RegressionObjective

public class Huber
extends Object
implements RegressionObjective

Huber loss, i.e., a mixture of l2 and l1 losses.

Constructor Summary

Constructors

Constructor and Description
Huber()
Huber(double cost)

Method Summary

Modifier and Type	Method and Description
com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance	getProvenance()
com.oracle.labs.mlrg.olcut.util.Pair<Double,SGDVector>	loss(DenseVector truth, SGDVector prediction) Scores a prediction, returning the loss.
void	postConfig() Used by the OLCUT configuration system, and should not be called by external code.
String	toString()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

Huber

public Huber()

Huber

public Huber(double cost)

Method Detail

postConfig

public void postConfig()

Used by the OLCUT configuration system, and should not be called by external code.

Specified by:

postConfig in interface com.oracle.labs.mlrg.olcut.config.Configurable

loss

public com.oracle.labs.mlrg.olcut.util.Pair<Double,SGDVector> loss(DenseVector truth,
                                                                   SGDVector prediction)

Description copied from interface: RegressionObjective

Scores a prediction, returning the loss.

Specified by:

loss in interface RegressionObjective

Parameters:

truth - The true regression value.

prediction - The predicted regression value.

Returns:

A pair with the loss and gradient.

toString

public String toString()

Overrides:

toString in class Object

getProvenance

public com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance getProvenance()

Specified by:

getProvenance in interface com.oracle.labs.mlrg.olcut.provenance.Provenancable<com.oracle.labs.mlrg.olcut.provenance.ConfiguredObjectProvenance>