Class Softmax

  • All Implemented Interfaces:
    java.io.Serializable

    public class Softmax
    extends Activation
    Softmax activation. Currently only works for output layer with cross entropy loss.
    See Also:
    Serialized Form
    • Constructor Summary

      Constructors 
      Constructor Description
      Softmax()  
    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      java.util.function.Consumer<Matrix> getActivation()
      Get a function that will apply the activation.
      java.util.function.Function<Matrix,​Matrix> getTransferDerivative()
      Get a function that will transfer derivative onto an already-activated matrix.
      java.lang.String toString()
      Get a string representation.
      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
    • Constructor Detail

      • Softmax

        public Softmax()
    • Method Detail

      • getActivation

        public java.util.function.Consumer<Matrix> getActivation()
        Description copied from class: Activation
        Get a function that will apply the activation.
        Specified by:
        getActivation in class Activation
        Returns:
        the function
      • getTransferDerivative

        public java.util.function.Function<Matrix,​Matrix> getTransferDerivative()
        Description copied from class: Activation
        Get a function that will transfer derivative onto an already-activated matrix.
        Specified by:
        getTransferDerivative in class Activation
        Returns:
        the function
      • toString

        public java.lang.String toString()
        Description copied from class: Activation
        Get a string representation.
        Specified by:
        toString in class Activation
        Returns:
        the string