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.String
toString()
Get a string representation.
-
-
-
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 classActivation
- 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 classActivation
- Returns:
- the function
-
toString
public java.lang.String toString()
Description copied from class:Activation
Get a string representation.- Specified by:
toString
in classActivation
- Returns:
- the string
-
-