Class GraphWiseBaseConvLayerConfigBuilder<Config extends GraphWiseBaseConvLayerConfig,​Self extends GraphWiseBaseConvLayerConfigBuilder>

    • Constructor Detail

      • GraphWiseBaseConvLayerConfigBuilder

        public GraphWiseBaseConvLayerConfigBuilder()
    • Method Detail

      • 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
      • 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