Class GraphWisePredictionLayerConfig

java.lang.Object
oracle.pgx.config.mllib.GraphWisePredictionLayerConfig
public class GraphWisePredictionLayerConfig extends Object
Configuration class for GraphWise prediction layers. See GraphWisePredictionLayerConfigBuilder for a description of the hyperparameters.
Since:
19.4

  • Field Details

  • Constructor Details

    • GraphWisePredictionLayerConfig

      public GraphWisePredictionLayerConfig()

  • Method Details

    • getHiddenDimension

      public Integer getHiddenDimension()

    • setHiddenDimension

      public void setHiddenDimension(Integer hiddenDimension)

    • getActivationFunction

      public ActivationFunction getActivationFunction()

    • setActivationFunction

      public void setActivationFunction(ActivationFunction activationFunction)

    • getWeightInitScheme

      public WeightInitScheme getWeightInitScheme()

    • setWeightInitScheme

      public void setWeightInitScheme(WeightInitScheme weightInitScheme)

    • getDropoutRate

      public double getDropoutRate()

    • setDropoutRate

      public void setDropoutRate(double dropoutRate)