oracle.pgx.api.mllib

Class GraphWiseAttentionLayerConfigBuilder

java.lang.Object

oracle.pgx.api.mllib.GraphWiseBaseConvLayerConfigBuilder<GraphWiseAttentionLayerConfig,GraphWiseAttentionLayerConfigBuilder>

oracle.pgx.api.mllib.GraphWiseAttentionLayerConfigBuilder

public class GraphWiseAttentionLayerConfigBuilder
extends GraphWiseBaseConvLayerConfigBuilder<GraphWiseAttentionLayerConfig,GraphWiseAttentionLayerConfigBuilder>

Builder for GraphWiseAttentionLayerConfig.

Since:

23.3

Constructor Summary

Constructors

Constructor and Description
GraphWiseAttentionLayerConfigBuilder()

Method Summary

Modifier and Type	Method and Description
GraphWiseAttentionLayerConfig	build() Build the configuration
GraphWiseAttentionLayerConfigBuilder	setHeadAggregation(AggregationOperation aggregationOperation) Set the aggregation operation for this layer (see AggregationOperation)
GraphWiseAttentionLayerConfigBuilder	setNumHeads(int numHeads) Set the number of heads that will be used in this layer.

Methods inherited from class oracle.pgx.api.mllib.GraphWiseBaseConvLayerConfigBuilder

setActivationFunction, setDropoutRate, setNumSampledNeighbors, setWeightInitScheme, useEdgeToEdgeConnection, useEdgeToVertexConnection, useVertexToEdgeConnection, useVertexToVertexConnection

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

GraphWiseAttentionLayerConfigBuilder

public GraphWiseAttentionLayerConfigBuilder()

Method Detail

build

public GraphWiseAttentionLayerConfig build()

Build the configuration

Returns:

built configuration

Since:

23.3

setHeadAggregation

public GraphWiseAttentionLayerConfigBuilder setHeadAggregation(AggregationOperation aggregationOperation)

Set the aggregation operation for this layer (see AggregationOperation)

default: GraphWiseAttentionLayerConfig.DEFAULT_AGGREGATION_OPERATION

Parameters:

aggregationOperation - aggregation operation

Returns:

this

Since:

23.3

setNumHeads

public GraphWiseAttentionLayerConfigBuilder setNumHeads(int numHeads)

Set the number of heads that will be used in this layer.

default: GraphWiseAttentionLayerConfig.DEFAULT_NUM_HEADS

Parameters:

numHeads - number of heads to use

Returns:

this

Since:

23.3