public class LossLayer extends FeedForwardLayer
Modifier and Type | Class and Description |
---|---|
static class |
LossLayer.Builder |
Modifier and Type | Field and Description |
---|---|
protected org.nd4j.linalg.lossfunctions.ILossFunction |
lossFn |
nIn, nOut
activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, layerName, learningRate, learningRateSchedule, momentum, momentumSchedule, rho, rmsDecay, updater, weightInit
Modifier | Constructor and Description |
---|---|
protected |
LossLayer(LossLayer.Builder builder) |
Modifier and Type | Method and Description |
---|---|
ParamInitializer |
initializer() |
Layer |
instantiate(NeuralNetConfiguration conf,
java.util.Collection<IterationListener> iterationListeners,
int layerIndex,
org.nd4j.linalg.api.ndarray.INDArray layerParamsView,
boolean initializeParams) |
getL1ByParam, getL2ByParam, getLearningRateByParam, getOutputType, getPreProcessorForInputType, setNIn
clone, getUpdaterByParam, resetLayerDefaultConfig
protected LossLayer(LossLayer.Builder builder)
public Layer instantiate(NeuralNetConfiguration conf, java.util.Collection<IterationListener> iterationListeners, int layerIndex, org.nd4j.linalg.api.ndarray.INDArray layerParamsView, boolean initializeParams)
instantiate
in class Layer
public ParamInitializer initializer()
initializer
in class Layer