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

Method Summary

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

public static final int DEFAULT_NUM_HEADS

3

See Also:

Constant Field Values

DEFAULT_AGGREGATION_OPERATION

public static final AggregationOperation DEFAULT_AGGREGATION_OPERATION

AggregationOperation.MEAN

Constructor Detail

GraphWiseAttentionLayerConfig

public GraphWiseAttentionLayerConfig()

GraphWiseAttentionLayerConfig

public GraphWiseAttentionLayerConfig(GraphWiseAttentionLayerConfig source)

Method Detail

getNumHeads

public int getNumHeads()

setNumHeads

public void setNumHeads(int numHeads)

getHeadAggregation

public AggregationOperation getHeadAggregation()

setHeadAggregation

public void setHeadAggregation(AggregationOperation aggregationOperation)