Package me.yixqiao.jlearn.activations
Class Softmax
- java.lang.Object
-
- me.yixqiao.jlearn.activations.Activation
-
- me.yixqiao.jlearn.activations.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.StringtoString()Get a string representation.
-
-
-
Method Detail
-
getActivation
public java.util.function.Consumer<Matrix> getActivation()
Description copied from class:ActivationGet a function that will apply the activation.- Specified by:
getActivationin classActivation- Returns:
- the function
-
getTransferDerivative
public java.util.function.Function<Matrix,Matrix> getTransferDerivative()
Description copied from class:ActivationGet a function that will transfer derivative onto an already-activated matrix.- Specified by:
getTransferDerivativein classActivation- Returns:
- the function
-
toString
public java.lang.String toString()
Description copied from class:ActivationGet a string representation.- Specified by:
toStringin classActivation- Returns:
- the string
-
-