public abstract class BasePretrainNetwork extends FeedForwardLayer
Modifier and Type | Class and Description |
---|---|
static class |
BasePretrainNetwork.Builder<T extends BasePretrainNetwork.Builder<T>> |
Modifier and Type | Field and Description |
---|---|
protected org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction |
lossFunction |
protected double |
visibleBiasInit |
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
Constructor and Description |
---|
BasePretrainNetwork(BasePretrainNetwork.Builder builder) |
Modifier and Type | Method and Description |
---|---|
double |
getL1ByParam(java.lang.String paramName)
Get the L1 coefficient for the given parameter.
|
double |
getL2ByParam(java.lang.String paramName)
Get the L2 coefficient for the given parameter.
|
double |
getLearningRateByParam(java.lang.String paramName)
Get the (initial) learning rate coefficient for the given parameter.
|
getOutputType, getPreProcessorForInputType, setNIn
clone, getUpdaterByParam, initializer, instantiate, resetLayerDefaultConfig
protected org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction
protected double visibleBiasInit
public BasePretrainNetwork(BasePretrainNetwork.Builder builder)
public double getL1ByParam(java.lang.String paramName)
Layer
getL1ByParam
in class FeedForwardLayer
paramName
- Parameter namepublic double getL2ByParam(java.lang.String paramName)
Layer
getL2ByParam
in class FeedForwardLayer
paramName
- Parameter namepublic double getLearningRateByParam(java.lang.String paramName)
Layer
getLearningRateByParam
in class FeedForwardLayer
paramName
- Parameter name