Interface LayerConstraint

    • Method Detail

      • applyConstraint

        void applyConstraint​(Layer layer,
                             int iteration,
                             int epoch)
        Apply a given constraint to a layer at each iteration in the provided epoch, after parameters have been updated.
        Parameters:
        layer - org.deeplearning4j.nn.api.Layer
        iteration - given iteration as integer
        epoch - current epoch as integer
      • setParams

        void setParams​(Set<String> params)
        Set the parameters that this layer constraint should be applied to
        Parameters:
        params - Parameters that the layer constraint should be applied to
      • getParams

        Set<String> getParams()
        Returns:
        Set of parameters that this layer constraint will be applied to