public abstract static class BasePretrainNetwork.Builder<T extends BasePretrainNetwork.Builder<T>> extends FeedForwardLayer.Builder<T>
Modifier and Type | Field and Description |
---|---|
protected org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction |
lossFunction |
protected int |
preTrainIterations |
protected double |
visibleBiasInit |
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() |
Modifier and Type | Method and Description |
---|---|
T |
lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction) |
T |
preTrainIterations(int preTrainIterations) |
T |
visibleBiasInit(double visibleBiasInit) |
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