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 java.lang.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 class	GraphWiseBaseModelConfig.Backend

Field Summary

Fields

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

Constructor Summary

Constructors

Constructor

Description

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

Methods inherited from class java.lang.Object

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

Field Detail

SUPPORTED_INPUT_TYPES

public static final java.util.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 java.lang.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 Detail

GraphWiseBaseModelConfig

public GraphWiseBaseModelConfig(int batchSize,
                                int numEpochs,
                                double learningRate,
                                double weightDecay,
                                int embeddingDim,
                                java.lang.Integer seed,
                                GraphWiseBaseConvLayerConfig[] convLayerConfigs,
                                boolean standardize,
                                boolean normalize,
                                boolean shuffle,
                                java.util.List<java.lang.String> vertexInputPropertyNames,
                                java.util.List<java.lang.String> edgeInputPropertyNames,
                                java.util.Map<java.lang.String,InputPropertyConfig> vertexInputPropertyConfigs,
                                java.util.Map<java.lang.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 Detail

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 java.lang.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 java.util.Map<java.lang.String,InputPropertyConfig> getVertexInputPropertyConfigs()

getEdgeInputPropertyConfigs

public java.util.Map<java.lang.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 java.util.List<java.lang.String> getVertexInputPropertyNames()

setVertexInputPropertyNames

public final void setVertexInputPropertyNames(java.util.List<java.lang.String> vertexInputPropertyNames)

getEdgeInputPropertyNames

public java.util.List<java.lang.String> getEdgeInputPropertyNames()

setEdgeInputPropertyNames

public final void setEdgeInputPropertyNames(java.util.List<java.lang.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()