Class GraphWiseBaseConvLayerConfigBuilder<Config extends GraphWiseBaseConvLayerConfig,Self extends GraphWiseBaseConvLayerConfigBuilder>

java.lang.Object
oracle.pgx.api.mllib.GraphWiseBaseConvLayerConfigBuilder<Config,Self>
Direct Known Subclasses:
GraphWiseAttentionLayerConfigBuilder, GraphWiseConvLayerConfigBuilder

public abstract class GraphWiseBaseConvLayerConfigBuilder<Config extends GraphWiseBaseConvLayerConfig,Self extends GraphWiseBaseConvLayerConfigBuilder> extends Object
Since:
23.3
  • Constructor Details

    • GraphWiseBaseConvLayerConfigBuilder

      public GraphWiseBaseConvLayerConfigBuilder()
  • Method Details

    • setNumSampledNeighbors

      public Self setNumSampledNeighbors(int numSampledNeighbors)
      Set the number of neighbors that will be sampled in this layer.

      default: GraphWiseBaseConvLayerConfig.DEFAULT_NUM_SAMPLED_NEIGHBORS

      Parameters:
      numSampledNeighbors - number of neighbors to sample
      Returns:
      this
      Since:
      19.4
    • setActivationFunction

      public Self setActivationFunction(ActivationFunction activationFunction)
      Set the activation function for this layer (see ActivationFunction)

      default: GraphWiseBaseConvLayerConfig.DEFAULT_ACTIVATION_FUNCTION

      Parameters:
      activationFunction - activation function
      Returns:
      this
      Since:
      19.4
    • setWeightInitScheme

      public Self setWeightInitScheme(WeightInitScheme weightInitScheme)
      Set the weight initialization scheme for the weights of this layer. Note that biases are always initialized to zeros. See WeightInitScheme.

      default: GraphWiseBaseConvLayerConfig.DEFAULT_WEIGHT_INIT_SCHEME

      Parameters:
      weightInitScheme - weight initialization scheme
      Returns:
      this
      Since:
      19.4
    • setDropoutRate

      public Self setDropoutRate(double dropoutRate)
      Set the dropout rate for this layer. Represents the probability to drop each neuron. Note that dropoutRate must satisfy 0 <= dropoutRate < 1. If dropoutRate = 0, dropout layer will not be constructed.

      default: GraphWiseBaseConvLayerConfig.DEFAULT_DROPOUT_RATE

      Parameters:
      dropoutRate - dropout rate
      Returns:
      this
      Since:
      22.3
    • useVertexToEdgeConnection

      public Self useVertexToEdgeConnection(boolean vertexToEdgeConnection)
      Use the connection between vertices and edges. Should be set to True if neighboring vertices should be sampled to compute the edges embeddings of that layer.

      default: GraphWiseBaseConvLayerConfig#DEFAULT_VERTEX_EDGE_CONNECTION

      Parameters:
      vertexToEdgeConnection - boolean flag to use vertex to edge connection
      Returns:
      this
      Since:
      22.2
    • useVertexToVertexConnection

      public Self useVertexToVertexConnection(boolean vertexToVertexConnection)
      Use the connection between vertices and vertices. Should be set to True if neighboring vertices should be sampled to compute the vertex embeddings of that layer.

      default: GraphWiseBaseConvLayerConfig#DEFAULT_VERTEX_VERTEX_CONNECTION

      Parameters:
      vertexToVertexConnection - boolean flag to use vertex to vertex connection
      Returns:
      this
      Since:
      22.2
    • useEdgeToVertexConnection

      public Self useEdgeToVertexConnection(boolean edgeToVertexConnection)
      Use the connection between edges and vertices. Should be set to True if neighboring edges should be sampled to compute the vertex embeddings of that layer.

      default: GraphWiseBaseConvLayerConfig#DEFAULT_EDGE_VERTEX_CONNECTION

      Parameters:
      edgeToVertexConnection - boolean flag to use edge to vertex connection
      Returns:
      this
      Since:
      22.2
    • useEdgeToEdgeConnection

      public Self useEdgeToEdgeConnection(boolean edgeToEdgeConnection)
      Use the connection between edges and edges. Should be set to True if neighboring edges should be sampled to compute the edge embeddings of that layer.

      default: GraphWiseBaseConvLayerConfig#DEFAULT_EDGE_EDGE_CONNECTION

      Parameters:
      edgeToEdgeConnection - boolean flag to use edge to edge connection
      Returns:
      this
      Since:
      22.2