public abstract static class BaseOutputLayer.Builder<T extends BaseOutputLayer.Builder<T>> extends FeedForwardLayer.Builder<T>
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, learningRatePolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, weightInit
Constructor and Description |
---|
Builder() |
Builder(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction) |
Builder(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction) |
Modifier and Type | Method and Description |
---|---|
T |
lossFunction(org.nd4j.linalg.lossfunctions.ILossFunction lossFunction) |
T |
lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction) |
nIn, nOut
activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, build, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, name, rho, rmsDecay, updater, weightInit