Package 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	Description
GraphWiseAttentionLayerConfigBuilder()

Method Summary

Modifier and Type	Method	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

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

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

build

public GraphWiseAttentionLayerConfig build()

Build the configuration

Returns:

built configuration

Since:

23.3