Class CrossEntropy

  • All Implemented Interfaces:
    java.io.Serializable

    public class CrossEntropy
    extends Loss
    Cross entropy loss.

    For now, only for use with the softmax activation.

    See Also:
    Serialized Form
    • Constructor Detail

      • CrossEntropy

        public CrossEntropy()
    • Method Detail

      • getLoss

        public double getLoss​(java.util.ArrayList<Matrix> out,
                              java.util.ArrayList<Matrix> y)
        Description copied from class: Loss
        Get the loss.
        Specified by:
        getLoss in class Loss
        Parameters:
        out - output of network
        y - correct output
        Returns:
        the loss