Class WeightNoise

    • Constructor Detail

      • WeightNoise

        public WeightNoise​(Distribution distribution)
        Parameters:
        distribution - Distribution for additive noise
      • WeightNoise

        public WeightNoise​(Distribution distribution,
                           boolean additive)
        Parameters:
        distribution - Distribution for noise
        additive - If true: noise is added to weights. If false: noise is multiplied by weights
      • WeightNoise

        public WeightNoise​(Distribution distribution,
                           boolean applyToBias,
                           boolean additive)
        Parameters:
        distribution - Distribution for noise
        applyToBias - If true: apply to biases also. If false (default): apply only to weights
        additive - If true: noise is added to weights. If false: noise is multiplied by weights
    • Method Detail

      • getParameter

        public INDArray getParameter​(Layer layer,
                                     String paramKey,
                                     int iteration,
                                     int epoch,
                                     boolean train,
                                     LayerWorkspaceMgr workspaceMgr)
        Description copied from interface: IWeightNoise
        Get the parameter, after applying weight noise
        Specified by:
        getParameter in interface IWeightNoise
        Parameters:
        layer - Layer to get the parameter for
        paramKey - Parameter key
        iteration - Iteration number
        epoch - Epoch number
        train - If true: training. False: at test time
        Returns:
        Parameter, after applying weight noise