org.ea.javacnn.losslayers

Class SoftMaxLayer

java.lang.Object

org.ea.javacnn.losslayers.LossLayer

org.ea.javacnn.losslayers.SoftMaxLayer

All Implemented Interfaces:

java.io.Serializable, Layer

public class SoftMaxLayer
extends LossLayer

This layer will squash the result of the activations in the fully connected layer and give you a value of 0 to 1 for all output activations.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
SoftMaxLayer(OutputDefinition def)

Method Summary

Modifier and Type	Method and Description
double	backward(int y)
DataBlock	forward(DataBlock db, boolean training)

Methods inherited from class org.ea.javacnn.losslayers.LossLayer

backward, getBackPropagationResult, getOutAct

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SoftMaxLayer

public SoftMaxLayer(OutputDefinition def)

Method Detail

forward

public DataBlock forward(DataBlock db,
                         boolean training)

backward

public double backward(int y)

Specified by:

backward in class LossLayer