oracle.pgx.api.mllib

Class GraphWisePredictionLayerConfigBuilder

java.lang.Object

oracle.pgx.api.mllib.GraphWisePredictionLayerConfigBuilder

public class GraphWisePredictionLayerConfigBuilder
extends java.lang.Object

Builder for GraphWisePredictionLayerConfig.

Since:

19.4

Constructor Summary

Constructors

Constructor and Description
GraphWisePredictionLayerConfigBuilder()

Method Summary

Modifier and Type	Method and Description
oracle.pgx.config.mllib.GraphWisePredictionLayerConfig	build() Build the configuration
GraphWisePredictionLayerConfigBuilder	setActivationFunction(oracle.pgx.config.mllib.ActivationFunction activationFunction) Set the activation function of the layer.
GraphWisePredictionLayerConfigBuilder	setHiddenDimension(java.lang.Integer hiddenDimension) Set the hidden dimension of the layer.
GraphWisePredictionLayerConfigBuilder	setWeightInitScheme(oracle.pgx.config.mllib.WeightInitScheme weightInitScheme) Sets the weight initialization scheme for the weights of the layer.

Methods inherited from class java.lang.Object

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

Constructor Detail

GraphWisePredictionLayerConfigBuilder

public GraphWisePredictionLayerConfigBuilder()

Method Detail

build

public oracle.pgx.config.mllib.GraphWisePredictionLayerConfig build()

Build the configuration

Returns:

built configuration

Since:

19.4

setActivationFunction

public GraphWisePredictionLayerConfigBuilder setActivationFunction(oracle.pgx.config.mllib.ActivationFunction activationFunction)

Set the activation function of the layer. See ActivationFunction.

Note that if this is the last layer, this setting will be ignored and replaced by the activation function of the loss function, e.g softmax or sigmoid.

default: GraphWisePredictionLayerConfig.DEFAULT_ACTIVATION_FUNCTION

Parameters:

activationFunction - activation function

Returns:

this

Since:

19.4

setHiddenDimension

public GraphWisePredictionLayerConfigBuilder setHiddenDimension(java.lang.Integer hiddenDimension)

Set the hidden dimension of the layer.

Note that if this is the last layer, this setting will be ignored and replaced by the number of classes

default: GraphWisePredictionLayerConfig.DEFAULT_HIDDEN_DIMENSION

Parameters:

hiddenDimension - hidden dimension

Returns:

this

Since:

19.4

setWeightInitScheme

public GraphWisePredictionLayerConfigBuilder setWeightInitScheme(oracle.pgx.config.mllib.WeightInitScheme weightInitScheme)

Sets the weight initialization scheme for the weights of the layer. Note that biases are always initialized with zeros. See WeightInitScheme.

default: GraphWisePredictionLayerConfig.DEFAULT_WEIGHT_INIT_SCHEME

Parameters:

weightInitScheme - weight initialization scheme

Returns:

this

Since:

19.4