oracle.pgx.api.mllib

Class GraphWiseDgiLayerConfigBuilder

java.lang.Object

oracle.pgx.api.mllib.GraphWiseDgiLayerConfigBuilder

public class GraphWiseDgiLayerConfigBuilder
extends java.lang.Object

Builder for GraphWiseDgiLayerConfig.

Since:

21.2

Constructor Summary

Constructors

Constructor and Description
GraphWiseDgiLayerConfigBuilder()

Method Summary

Modifier and Type	Method and Description
GraphWiseDgiLayerConfig	build() Build the configuration
GraphWiseDgiLayerConfigBuilder	setCorruptionFunction(CorruptionFunction corruptionFunction) Sets the corruption function of the dgi layer
GraphWiseDgiLayerConfigBuilder	setDiscriminator(GraphWiseDgiLayerConfig.Discriminator discriminator) Sets the discriminator of the dgi layer
GraphWiseDgiLayerConfigBuilder	setReadoutFunction(GraphWiseDgiLayerConfig.ReadoutFunction readoutFunction) Sets the readout function of the dgi layer

Methods inherited from class java.lang.Object

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

Constructor Detail

GraphWiseDgiLayerConfigBuilder

public GraphWiseDgiLayerConfigBuilder()

Method Detail

build

public GraphWiseDgiLayerConfig build()

Build the configuration

Returns:

built configuration

Since:

21.2

setCorruptionFunction

public GraphWiseDgiLayerConfigBuilder setCorruptionFunction(CorruptionFunction corruptionFunction)

Sets the corruption function of the dgi layer

default: GraphWiseDgiLayerConfig.DEFAULT_CORRUPTION_FUNCTION

Parameters:

corruptionFunction - corruption function of the dgi layer

Returns:

this

Since:

21.2

setDiscriminator

public GraphWiseDgiLayerConfigBuilder setDiscriminator(GraphWiseDgiLayerConfig.Discriminator discriminator)

Sets the discriminator of the dgi layer

default: GraphWiseDgiLayerConfig.DEFAULT_DISCRIMINATOR

Parameters:

discriminator - discriminator of the dgi layer

Returns:

this

Since:

21.2

setReadoutFunction

public GraphWiseDgiLayerConfigBuilder setReadoutFunction(GraphWiseDgiLayerConfig.ReadoutFunction readoutFunction)

Sets the readout function of the dgi layer

default: GraphWiseDgiLayerConfig.DEFAULT_READOUT_FUNCTION

Parameters:

readoutFunction - readout function of the dgi layer

Returns:

this

Since:

21.2