Package oracle.pgx.config.mllib

Class GraphWiseBaseModelConfig

java.lang.Object

oracle.pgx.config.mllib.GraphWiseBaseModelConfig

Direct Known Subclasses:

EdgeWiseModelConfig, GraphWiseModelConfig

public abstract class GraphWiseBaseModelConfig
extends Object

Abstract base configuration class for both GraphWise and EdgeWise models

Since:

23.1

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	GraphWiseBaseModelConfig.Backend

Field Summary

Fields

Modifier and Type	Field	Description
static final GraphWiseBaseModelConfig.Backend	DEFAULT_BACKEND	LibTorch
static final int	DEFAULT_BATCH_SIZE	128
static final GraphWiseBaseConvLayerConfig[]	DEFAULT_CONV_LAYER_CONFIGS	two default initialized layer configs (See GraphWiseBaseConvLayerConfig)
static final int	DEFAULT_EMBEDDING_DIM	128
static final boolean	DEFAULT_ENABLE_ACCELERATOR	true
static final double	DEFAULT_LEARNING_RATE	0.01
static final boolean	DEFAULT_NORMALIZE	true
static final int	DEFAULT_NUM_EPOCHS	3
static final Integer	DEFAULT_SEED	null
static final boolean	DEFAULT_SHUFFLE	true
static final boolean	DEFAULT_STANDARDIZE	false
static final double	DEFAULT_WEIGHT_DECAY	0.0
static final EnumSet<PropertyType>	SUPPORTED_INPUT_TYPES	BOOLEAN, FLOAT, DOUBLE

Constructor Summary

Constructors

Constructor

Description

GraphWiseBaseModelConfig(int batchSize, int numEpochs, double learningRate, double weightDecay, int embeddingDim, Integer seed, GraphWiseBaseConvLayerConfig[] convLayerConfigs, boolean standardize, boolean normalize, boolean shuffle, List<String> vertexInputPropertyNames, List<String> edgeInputPropertyNames, Map<String,InputPropertyConfig> vertexInputPropertyConfigs, Map<String,InputPropertyConfig> edgeInputPropertyConfigs, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig, boolean fitted, double trainingLoss, int vertexInputFeatureDim, int edgeInputFeatureDim, GraphWiseBaseModelConfig.Backend backend, boolean enableAccelerator, GraphWiseValidationConfig validationConfig)

Method Summary

Modifier and Type	Method	Description
final void	clearSensitiveData()
GraphWiseBaseModelConfig.Backend	getBackend()
int	getBatchSize()
oracle.pgx.config.internal.categorymapping.CategoryMappingConfig	getCategoryMappingConfig()
GraphWiseBaseConvLayerConfig[]	getConvLayerConfigs()
int	getEdgeInputFeatureDim()
Map<String,InputPropertyConfig>	getEdgeInputPropertyConfigs()
List<String>	getEdgeInputPropertyNames()
int	getEmbeddingDim()
int	getInputFeatureDim()
double	getLearningRate()
int	getNumEpochs()
Integer	getSeed()
double	getTrainingLoss()
GraphWiseValidationConfig	getValidationConfig()
Map<String,InputPropertyConfig>	getVertexInputPropertyConfigs()
List<String>	getVertexInputPropertyNames()
double	getWeightDecay()
boolean	isEnableAccelerator()
boolean	isFitted()
boolean	isNormalize()
boolean	isShuffle()
boolean	isStandardize()
final void	setBatchSize(int batchSize)
final void	setCategoryMappingConfig(oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
final void	setConvLayerConfigs(GraphWiseBaseConvLayerConfig... convLayerConfigs)
final void	setEdgeInputFeatureDim(int edgeInputFeatureDim)
final void	setEdgeInputPropertyConfigs(InputPropertyConfig... edgeInputPropertyConfigs)
final void	setEdgeInputPropertyNames(List<String> edgeInputPropertyNames)
final void	setEmbeddingDim(int embeddingDim)
void	setEnableAccelerator(boolean enableAccelerator)
final void	setFitted(boolean fitted)
final void	setInputFeatureDim(int vertexInputFeatureDim)
final void	setLearningRate(double learningRate)
final void	setNormalize(boolean normalize)
final void	setNumEpochs(int numEpochs)
final void	setSeed(int seed)
final void	setShuffle(boolean shuffle)
final void	setStandardize(boolean standardize)
final void	setTrainingLoss(double trainingLoss)
final void	setValidationConfig(GraphWiseValidationConfig validationConfig)
final void	setVertexInputPropertyConfigs(InputPropertyConfig... vertexInputPropertyConfigs)
final void	setVertexInputPropertyNames(List<String> vertexInputPropertyNames)
final void	setWeightDecay(double weightDecay)

Methods inherited from class java.lang.Object

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

Field Details

SUPPORTED_INPUT_TYPES

public static final EnumSet<PropertyType> SUPPORTED_INPUT_TYPES

BOOLEAN, FLOAT, DOUBLE

DEFAULT_BATCH_SIZE

public static final int DEFAULT_BATCH_SIZE

128

See Also:

• Constant Field Values

DEFAULT_NUM_EPOCHS

public static final int DEFAULT_NUM_EPOCHS

3

See Also:

• Constant Field Values

DEFAULT_LEARNING_RATE

public static final double DEFAULT_LEARNING_RATE

0.01

See Also:

• Constant Field Values

DEFAULT_WEIGHT_DECAY

public static final double DEFAULT_WEIGHT_DECAY

0.0

See Also:

• Constant Field Values

DEFAULT_EMBEDDING_DIM

public static final int DEFAULT_EMBEDDING_DIM

128

See Also:

• Constant Field Values

DEFAULT_SEED

public static final Integer DEFAULT_SEED

null

DEFAULT_BACKEND

public static final GraphWiseBaseModelConfig.Backend DEFAULT_BACKEND

LibTorch

DEFAULT_CONV_LAYER_CONFIGS

public static final GraphWiseBaseConvLayerConfig[] DEFAULT_CONV_LAYER_CONFIGS

two default initialized layer configs (See GraphWiseBaseConvLayerConfig)

DEFAULT_STANDARDIZE

public static final boolean DEFAULT_STANDARDIZE

false

See Also:

• Constant Field Values

DEFAULT_NORMALIZE

public static final boolean DEFAULT_NORMALIZE

true

See Also:

• Constant Field Values

DEFAULT_SHUFFLE

public static final boolean DEFAULT_SHUFFLE

true

See Also:

• Constant Field Values

DEFAULT_ENABLE_ACCELERATOR

public static final boolean DEFAULT_ENABLE_ACCELERATOR

true

See Also:

• Constant Field Values

Constructor Details

GraphWiseBaseModelConfig

public GraphWiseBaseModelConfig(int batchSize,
 int numEpochs,
 double learningRate,
 double weightDecay,
 int embeddingDim,
 Integer seed,
 GraphWiseBaseConvLayerConfig[] convLayerConfigs,
 boolean standardize,
 boolean normalize,
 boolean shuffle,
 List<String> vertexInputPropertyNames,
 List<String> edgeInputPropertyNames,
 Map<String,InputPropertyConfig> vertexInputPropertyConfigs,
 Map<String,InputPropertyConfig> edgeInputPropertyConfigs,
 oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig,
 boolean fitted,
 double trainingLoss,
 int vertexInputFeatureDim,
 int edgeInputFeatureDim,
 GraphWiseBaseModelConfig.Backend backend,
 boolean enableAccelerator,
 GraphWiseValidationConfig validationConfig)

Method Details

isShuffle

public boolean isShuffle()

setShuffle

public final void setShuffle(boolean shuffle)

getInputFeatureDim

public int getInputFeatureDim()

setInputFeatureDim

public final void setInputFeatureDim(int vertexInputFeatureDim)

getEdgeInputFeatureDim

public int getEdgeInputFeatureDim()

setEdgeInputFeatureDim

public final void setEdgeInputFeatureDim(int edgeInputFeatureDim)

isFitted

public boolean isFitted()

setFitted

public final void setFitted(boolean fitted)

getTrainingLoss

public double getTrainingLoss()

setTrainingLoss

public final void setTrainingLoss(double trainingLoss)

getBatchSize

public int getBatchSize()

setBatchSize

public final void setBatchSize(int batchSize)

getNumEpochs

public int getNumEpochs()

setNumEpochs

public final void setNumEpochs(int numEpochs)

getLearningRate

public double getLearningRate()

setLearningRate

public final void setLearningRate(double learningRate)

getWeightDecay

public double getWeightDecay()

setWeightDecay

public final void setWeightDecay(double weightDecay)

getEmbeddingDim

public int getEmbeddingDim()

setEmbeddingDim

public final void setEmbeddingDim(int embeddingDim)

getSeed

public Integer getSeed()

setSeed

public final void setSeed(int seed)

getConvLayerConfigs

public GraphWiseBaseConvLayerConfig[] getConvLayerConfigs()

setConvLayerConfigs

public final void setConvLayerConfigs(GraphWiseBaseConvLayerConfig... convLayerConfigs)

getValidationConfig

public GraphWiseValidationConfig getValidationConfig()

setValidationConfig

public final void setValidationConfig(GraphWiseValidationConfig validationConfig)

getVertexInputPropertyConfigs

public Map<String,InputPropertyConfig> getVertexInputPropertyConfigs()

getEdgeInputPropertyConfigs

public Map<String,InputPropertyConfig> getEdgeInputPropertyConfigs()

setVertexInputPropertyConfigs

public final void setVertexInputPropertyConfigs(InputPropertyConfig... vertexInputPropertyConfigs)

setEdgeInputPropertyConfigs

public final void setEdgeInputPropertyConfigs(InputPropertyConfig... edgeInputPropertyConfigs)

getCategoryMappingConfig

public oracle.pgx.config.internal.categorymapping.CategoryMappingConfig getCategoryMappingConfig()

setCategoryMappingConfig

public final void setCategoryMappingConfig(oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)

clearSensitiveData

public final void clearSensitiveData()

getVertexInputPropertyNames

public List<String> getVertexInputPropertyNames()

setVertexInputPropertyNames

public final void setVertexInputPropertyNames(List<String> vertexInputPropertyNames)

getEdgeInputPropertyNames

public List<String> getEdgeInputPropertyNames()

setEdgeInputPropertyNames

public final void setEdgeInputPropertyNames(List<String> edgeInputPropertyNames)

isStandardize

public boolean isStandardize()

setStandardize

public final void setStandardize(boolean standardize)

isNormalize

public boolean isNormalize()

setNormalize

public final void setNormalize(boolean normalize)

getBackend

public GraphWiseBaseModelConfig.Backend getBackend()

setEnableAccelerator

public void setEnableAccelerator(boolean enableAccelerator)

isEnableAccelerator

public boolean isEnableAccelerator()