Class GraphWisePredictionLayerConfig


  • public class GraphWisePredictionLayerConfig
    extends java.lang.Object
    Configuration class for GraphWise prediction layers. See GraphWisePredictionLayerConfigBuilder for a description of the hyperparameters.
    Since:
    19.4
    • Constructor Detail

      • GraphWisePredictionLayerConfig

        public GraphWisePredictionLayerConfig()
    • Method Detail

      • getHiddenDimension

        public java.lang.Integer getHiddenDimension()
      • setHiddenDimension

        public void setHiddenDimension​(java.lang.Integer hiddenDimension)
      • setActivationFunction

        public void setActivationFunction​(ActivationFunction activationFunction)
      • setWeightInitScheme

        public void setWeightInitScheme​(WeightInitScheme weightInitScheme)
      • getDropoutRate

        public double getDropoutRate()
      • setDropoutRate

        public void setDropoutRate​(double dropoutRate)