oracle.pgx.config.mllib

Class GraphWiseAttentionLayerConfig

java.lang.Object

oracle.pgx.config.mllib.GraphWiseBaseConvLayerConfig

oracle.pgx.config.mllib.GraphWiseAttentionLayerConfig

public class GraphWiseAttentionLayerConfig
extends GraphWiseBaseConvLayerConfig

Configuration class for GraphWise attention layers. See GraphWiseAttentionLayerConfigBuilder for a description of the hyperparameters.

Since:

23.3

Nested Class Summary

Nested classes/interfaces inherited from class oracle.pgx.config.mllib.GraphWiseBaseConvLayerConfig

GraphWiseBaseConvLayerConfig.ConvLayerType

Field Summary

Fields

Modifier and Type	Field and Description
static AggregationOperation	DEFAULT_AGGREGATION_OPERATION AggregationOperation.MEAN
static int	DEFAULT_NUM_HEADS 3

Fields inherited from class oracle.pgx.config.mllib.GraphWiseBaseConvLayerConfig

DEFAULT_ACTIVATION_FUNCTION, DEFAULT_DROPOUT_RATE, DEFAULT_NUM_SAMPLED_NEIGHBORS, DEFAULT_WEIGHT_INIT_SCHEME

Constructor Summary

Constructors

Constructor and Description
GraphWiseAttentionLayerConfig()
GraphWiseAttentionLayerConfig(GraphWiseAttentionLayerConfig source)

Method Summary

Modifier and Type	Method and Description
AggregationOperation	getHeadAggregation()
int	getNumHeads()
void	setHeadAggregation(AggregationOperation aggregationOperation)
void	setNumHeads(int numHeads)

Methods inherited from class oracle.pgx.config.mllib.GraphWiseBaseConvLayerConfig

getActivationFunction, getConvLayerType, getDropoutRate, getEdgeToEdgeConnection, getEdgeToVertexConnection, getNeighborWeightPropertyName, getNumSampledNeighbors, getVertexToEdgeConnection, getVertexToVertexConnection, getWeightInitScheme, setActivationFunction, setDropoutRate, setNumSampledNeighbors, setWeightInitScheme, useEdgeToEdgeConnection, useEdgeToVertexConnection, useVertexToEdgeConnection, useVertexToVertexConnection

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_AGGREGATION_OPERATION

public static final AggregationOperation DEFAULT_AGGREGATION_OPERATION

AggregationOperation.MEAN

DEFAULT_NUM_HEADS

public static final int DEFAULT_NUM_HEADS

3

See Also:

Constant Field Values

Constructor Detail

GraphWiseAttentionLayerConfig

public GraphWiseAttentionLayerConfig()

GraphWiseAttentionLayerConfig

public GraphWiseAttentionLayerConfig(GraphWiseAttentionLayerConfig source)

Method Detail

getHeadAggregation

public AggregationOperation getHeadAggregation()

getNumHeads

public int getNumHeads()

setHeadAggregation

public void setHeadAggregation(AggregationOperation aggregationOperation)

setNumHeads

public void setNumHeads(int numHeads)