public static class RBM.Builder extends BasePretrainNetwork.Builder<RBM.Builder>
lossFunction, preTrainIterations, 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() |
Builder(RBM.HiddenUnit hiddenUnit,
RBM.VisibleUnit visibleUnit) |
Modifier and Type | Method and Description |
---|---|
RBM |
build() |
RBM.Builder |
hiddenUnit(RBM.HiddenUnit hiddenUnit) |
RBM.Builder |
k(int k) |
RBM.Builder |
sparsity(double sparsity) |
RBM.Builder |
visibleUnit(RBM.VisibleUnit visibleUnit) |
lossFunction, preTrainIterations, visibleBiasInit
nIn, nOut
activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, name, rho, rmsDecay, updater, weightInit
public Builder(RBM.HiddenUnit hiddenUnit, RBM.VisibleUnit visibleUnit)
public Builder()
public RBM build()
build
in class Layer.Builder<RBM.Builder>
public RBM.Builder k(int k)
public RBM.Builder hiddenUnit(RBM.HiddenUnit hiddenUnit)
public RBM.Builder visibleUnit(RBM.VisibleUnit visibleUnit)
public RBM.Builder sparsity(double sparsity)