public class GraphWiseDgiLayerConfigBuilder
extends java.lang.Object
GraphWiseDgiLayerConfig
.Constructor and Description |
---|
GraphWiseDgiLayerConfigBuilder() |
Modifier and Type | Method and Description |
---|---|
oracle.pgx.config.mllib.GraphWiseDgiLayerConfig |
build()
Build the configuration
|
GraphWiseDgiLayerConfigBuilder |
setCorruptionFunction(oracle.pgx.config.mllib.corruption.CorruptionFunction corruptionFunction)
Sets the corruption function of the dgi layer
|
GraphWiseDgiLayerConfigBuilder |
setDiscriminator(oracle.pgx.config.mllib.GraphWiseDgiLayerConfig.Discriminator discriminator)
Sets the discriminator of the dgi layer
|
GraphWiseDgiLayerConfigBuilder |
setReadoutFunction(oracle.pgx.config.mllib.GraphWiseDgiLayerConfig.ReadoutFunction readoutFunction)
Sets the readout function of the dgi layer
|
public oracle.pgx.config.mllib.GraphWiseDgiLayerConfig build()
public GraphWiseDgiLayerConfigBuilder setCorruptionFunction(oracle.pgx.config.mllib.corruption.CorruptionFunction corruptionFunction)
default: GraphWiseDgiLayerConfig.DEFAULT_CORRUPTION_FUNCTION
corruptionFunction
- corruption function of the dgi layerpublic GraphWiseDgiLayerConfigBuilder setDiscriminator(oracle.pgx.config.mllib.GraphWiseDgiLayerConfig.Discriminator discriminator)
default: GraphWiseDgiLayerConfig.DEFAULT_DISCRIMINATOR
discriminator
- discriminator of the dgi layerpublic GraphWiseDgiLayerConfigBuilder setReadoutFunction(oracle.pgx.config.mllib.GraphWiseDgiLayerConfig.ReadoutFunction readoutFunction)
default: GraphWiseDgiLayerConfig.DEFAULT_READOUT_FUNCTION
readoutFunction
- readout function of the dgi layer