Class GraphWiseDominantLayerConfigBuilder

    • Constructor Detail

      • GraphWiseDominantLayerConfigBuilder

        public GraphWiseDominantLayerConfigBuilder()
    • Method Detail

      • setDecoderLayerConfigs

        public GraphWiseDominantLayerConfigBuilder setDecoderLayerConfigs​(GraphWisePredictionLayerConfig... layerConfigs)
        Set the decoder layer configurations (See GraphWisePredictionLayerConfig. You must pass at least one prediction layer config (or leave it as default).

        Note that an additional layer will be inserted in the end for which:

        • the hidden dimension will be equal to the input feature dimension
        • the weight initialization scheme will be copied from the previous layer

        default: null

        Parameters:
        layerConfigs - layer configs
        Returns:
        this
        Since:
        23.2
      • setAlpha

        public GraphWiseDominantLayerConfigBuilder setAlpha​(double alpha)
        Set the alpha weight, the proportion of importance of feature reconstruction over graph structure reconstruction
        Parameters:
        alpha - feature reconstruction weights
        Returns:
        this
        Since:
        23.2