Package oracle.pgx.config.mllib

Class SupervisedEdgeWiseModelConfig

java.lang.Object

oracle.pgx.config.mllib.GraphWiseBaseModelConfig

oracle.pgx.config.mllib.EdgeWiseModelConfig

oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig

public class SupervisedEdgeWiseModelConfig
extends EdgeWiseModelConfig

Configuration class for SupervisedEdgeWiseModel. See SupervisedEdgeWiseModel for a description of the hyperparameters.

Since:

23.1

Nested Class Summary

Nested classes/interfaces inherited from class oracle.pgx.config.mllib.EdgeWiseModelConfig

EdgeWiseModelConfig.EdgeWiseConvModelVariant

Nested classes/interfaces inherited from class oracle.pgx.config.mllib.GraphWiseBaseModelConfig

GraphWiseBaseModelConfig.Backend

Field Summary

Fields

Modifier and Type	Field	Description
static BatchGenerator	DEFAULT_BATCH_GENERATOR	StandardBatchGenerator
static java.util.Map<?,java.lang.Float>	DEFAULT_CLASS_MAP	null
static LossFunction	DEFAULT_LOSS_FUNCTION_CLASS	SoftmaxCrossEntropyLoss
static GraphWisePredictionLayerConfig[]	DEFAULT_PREDICTION_LAYER_CONFIGS	one default initialized config (See GraphWisePredictionLayerConfig)
static java.util.EnumSet<PropertyType>	SUPPORTED_LABEL_TYPES	INTEGER, STRING, BOOLEAN, LONG
static java.util.EnumSet<PropertyType>	SUPPORTED_REGRESSION_TYPES	FLOAT, DOUBLE, INTEGER, LONG

Fields inherited from class oracle.pgx.config.mllib.EdgeWiseModelConfig

DEFAULT_MODE, EDGE_EMBEDDING_SIZE

Fields inherited from class oracle.pgx.config.mllib.GraphWiseBaseModelConfig

DEFAULT_BACKEND, DEFAULT_BATCH_SIZE, DEFAULT_CONV_LAYER_CONFIGS, DEFAULT_EMBEDDING_DIM, DEFAULT_ENABLE_ACCELERATOR, DEFAULT_LEARNING_RATE, DEFAULT_NORMALIZE, DEFAULT_NUM_EPOCHS, DEFAULT_SEED, DEFAULT_SHUFFLE, DEFAULT_STANDARDIZE, DEFAULT_WEIGHT_DECAY, SUPPORTED_INPUT_TYPES

Constructor Summary

Constructors

Constructor	Description
SupervisedEdgeWiseModelConfig()
SupervisedEdgeWiseModelConfig(int batchSize, int numEpochs, double learningRate, double weightDecay, int embeddingDim, java.lang.Integer edgeEmbeddingDim, java.lang.Integer seed, GraphWiseBaseConvLayerConfig[] convLayerConfigs, boolean standardize, 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, java.util.List<java.util.Set<java.lang.String>> targetEdgeLabelSets, boolean fitted, double trainingLoss, int inputFeatureDim, int edgeInputFeatureDim, oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction lossFunction, LossFunction lossFunctionClass, BatchGenerator batchGenerator, GraphWisePredictionLayerConfig[] predictionLayerConfigs, boolean normalize, java.lang.String edgeTargetPropertyName, LabelMaps labelMaps, GraphWiseBaseModelConfig.Backend backend, EdgeCombinationMethod edgeCombinationMethod, EdgeWiseModelConfig.EdgeWiseConvModelVariant variant, boolean enableAccelerator, GraphWiseValidationConfig validationConfig)
SupervisedEdgeWiseModelConfig(SupervisedEdgeWiseModelConfig source)
SupervisedEdgeWiseModelConfig(SupervisedEdgeWiseModelConfig source, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)

Method Summary

Modifier and Type	Method	Description
BatchGenerator	getBatchGenerator()
java.util.Map<?,java.lang.Integer>	getClassMap()
java.util.Map<?,java.lang.Float>	getClassWeights()
java.lang.String	getEdgeTargetPropertyName()
LabelMaps	getLabelMaps()
PropertyType	getLabelType()
LossFunction	getLossFunctionClass()
int	getNumClasses()
GraphWisePredictionLayerConfig[]	getPredictionLayerConfigs()
void	setBatchGenerator(BatchGenerator batchGenerator)
void	setClassMap(java.util.Map<?,java.lang.Integer> classMap)
void	setClassWeights(java.util.Map<?,java.lang.Float> classWeights)
void	setEdgeTargetPropertyName(java.lang.String edgeTargetPropertyName)
void	setLabelMaps(LabelMaps labelMaps)
void	setLabelType(PropertyType labelType)
void	setLossFunctionClass(LossFunction lossFunction)
void	setPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs)

Methods inherited from class oracle.pgx.config.mllib.EdgeWiseModelConfig

getEdgeCombinationMethod, getEdgeEmbeddingDim, getTargetEdgeLabelSets, getVariant, setEdgeCombinationMethod, setEdgeEmbeddingDim, setTargetEdgeLabels, setTargetEdgeLabelSets, setVariant

Methods inherited from class oracle.pgx.config.mllib.GraphWiseBaseModelConfig

clearSensitiveData, getBackend, getBatchSize, getCategoryMappingConfig, getConvLayerConfigs, getEdgeInputFeatureDim, getEdgeInputPropertyConfigs, getEdgeInputPropertyNames, getEmbeddingDim, getInputFeatureDim, getLearningRate, getNumEpochs, getSeed, getTrainingLoss, getValidationConfig, getVertexInputPropertyConfigs, getVertexInputPropertyNames, getWeightDecay, isEnableAccelerator, isFitted, isNormalize, isShuffle, isStandardize, setBatchSize, setCategoryMappingConfig, setConvLayerConfigs, setEdgeInputFeatureDim, setEdgeInputPropertyConfigs, setEdgeInputPropertyNames, setEmbeddingDim, setEnableAccelerator, setFitted, setInputFeatureDim, setLearningRate, setNormalize, setNumEpochs, setSeed, setShuffle, setStandardize, setTrainingLoss, setValidationConfig, setVertexInputPropertyConfigs, setVertexInputPropertyNames, setWeightDecay

Methods inherited from class java.lang.Object

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

Field Detail

SUPPORTED_LABEL_TYPES

public static final java.util.EnumSet<PropertyType> SUPPORTED_LABEL_TYPES

INTEGER, STRING, BOOLEAN, LONG

SUPPORTED_REGRESSION_TYPES

public static final java.util.EnumSet<PropertyType> SUPPORTED_REGRESSION_TYPES

FLOAT, DOUBLE, INTEGER, LONG

DEFAULT_LOSS_FUNCTION_CLASS

public static final LossFunction DEFAULT_LOSS_FUNCTION_CLASS

SoftmaxCrossEntropyLoss

DEFAULT_BATCH_GENERATOR

public static final BatchGenerator DEFAULT_BATCH_GENERATOR

StandardBatchGenerator

DEFAULT_PREDICTION_LAYER_CONFIGS

public static final GraphWisePredictionLayerConfig[] DEFAULT_PREDICTION_LAYER_CONFIGS

one default initialized config (See GraphWisePredictionLayerConfig)

DEFAULT_CLASS_MAP

public static final java.util.Map<?,java.lang.Float> DEFAULT_CLASS_MAP

null

Constructor Detail

SupervisedEdgeWiseModelConfig

public SupervisedEdgeWiseModelConfig()

SupervisedEdgeWiseModelConfig

public SupervisedEdgeWiseModelConfig(int batchSize,
                                     int numEpochs,
                                     double learningRate,
                                     double weightDecay,
                                     int embeddingDim,
                                     java.lang.Integer edgeEmbeddingDim,
                                     java.lang.Integer seed,
                                     GraphWiseBaseConvLayerConfig[] convLayerConfigs,
                                     boolean standardize,
                                     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,
                                     java.util.List<java.util.Set<java.lang.String>> targetEdgeLabelSets,
                                     boolean fitted,
                                     double trainingLoss,
                                     int inputFeatureDim,
                                     int edgeInputFeatureDim,
                                     oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction lossFunction,
                                     LossFunction lossFunctionClass,
                                     BatchGenerator batchGenerator,
                                     GraphWisePredictionLayerConfig[] predictionLayerConfigs,
                                     boolean normalize,
                                     java.lang.String edgeTargetPropertyName,
                                     LabelMaps labelMaps,
                                     GraphWiseBaseModelConfig.Backend backend,
                                     EdgeCombinationMethod edgeCombinationMethod,
                                     EdgeWiseModelConfig.EdgeWiseConvModelVariant variant,
                                     boolean enableAccelerator,
                                     GraphWiseValidationConfig validationConfig)

SupervisedEdgeWiseModelConfig

public SupervisedEdgeWiseModelConfig(SupervisedEdgeWiseModelConfig source)

SupervisedEdgeWiseModelConfig

public SupervisedEdgeWiseModelConfig(SupervisedEdgeWiseModelConfig source,
                                     oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)

Method Detail

getNumClasses

public int getNumClasses()

getEdgeTargetPropertyName

public java.lang.String getEdgeTargetPropertyName()

setEdgeTargetPropertyName

public final void setEdgeTargetPropertyName(java.lang.String edgeTargetPropertyName)

getPredictionLayerConfigs

public GraphWisePredictionLayerConfig[] getPredictionLayerConfigs()

setPredictionLayerConfigs

public final void setPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs)

getLossFunctionClass

public LossFunction getLossFunctionClass()

setLossFunctionClass

public final void setLossFunctionClass(LossFunction lossFunction)

getBatchGenerator

public BatchGenerator getBatchGenerator()

setBatchGenerator

public final void setBatchGenerator(BatchGenerator batchGenerator)

getClassMap

public java.util.Map<?,java.lang.Integer> getClassMap()

setClassMap

public final void setClassMap(java.util.Map<?,java.lang.Integer> classMap)

setClassWeights

public final void setClassWeights(java.util.Map<?,java.lang.Float> classWeights)

getClassWeights

public java.util.Map<?,java.lang.Float> getClassWeights()

getLabelType

public PropertyType getLabelType()

setLabelType

public final void setLabelType(PropertyType labelType)

getLabelMaps

public LabelMaps getLabelMaps()

setLabelMaps

public final void setLabelMaps(LabelMaps labelMaps)