org.apache.spark.ml.ann

Interface LossFunction

public interface LossFunction

Trait for loss function

Method Summary

Modifier and Type	Method and Description
double	loss(breeze.linalg.DenseMatrix<Object> output, breeze.linalg.DenseMatrix<Object> target, breeze.linalg.DenseMatrix<Object> delta) Returns the value of loss function.

Method Detail

loss

double loss(breeze.linalg.DenseMatrix<Object> output,
            breeze.linalg.DenseMatrix<Object> target,
            breeze.linalg.DenseMatrix<Object> delta)

Returns the value of loss function. Computes loss based on target and output. Writes delta (error) to delta in place. Delta is allocated based on the outputSize of model implementation.

Parameters:

output - actual output

target - target output

delta - delta (updated in place)

Returns:

loss