org.apache.reef.examples.group.bgd.loss

Class SquaredErrorLossFunction

java.lang.Object

org.apache.reef.examples.group.bgd.loss.SquaredErrorLossFunction

All Implemented Interfaces:

LossFunction

public class SquaredErrorLossFunction
extends Object
implements LossFunction

The Squared Error LossFunction.

Constructor Summary

Constructors

Constructor and Description
SquaredErrorLossFunction() Trivial constructor.

Method Summary

Methods

Modifier and Type	Method and Description
double	computeGradient(double y, double f) Computes the gradient with respect to f, if y is the true label.
double	computeLoss(double y, double f) Computes the loss incurred by predicting f, if y is the true label.
String	toString()

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

SquaredErrorLossFunction

@Inject
public SquaredErrorLossFunction()

Trivial constructor.

Method Detail

computeLoss

public double computeLoss(double y,
                 double f)

Description copied from interface: LossFunction

Computes the loss incurred by predicting f, if y is the true label.

Specified by:

computeLoss in interface LossFunction

Parameters:

y - the label

f - the prediction

Returns:

the loss incurred by predicting f, if y is the true label.

computeGradient

public double computeGradient(double y,
                     double f)

Description copied from interface: LossFunction

Computes the gradient with respect to f, if y is the true label.

Specified by:

computeGradient in interface LossFunction

Parameters:

y - the label

f - the prediction

Returns:

the gradient with respect to f

toString

public String toString()

Overrides:

toString in class Object