Package org.apache.spark.ml.ann

Interface LossFunction

public interface LossFunction
Trait for loss function

  • Method Summary

    Modifier and Type
    Method
    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 Details

    • 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