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 final BatchGenerator	DEFAULT_BATCH_GENERATOR	StandardBatchGenerator
static final Map<?,Float>	DEFAULT_CLASS_MAP	null
static final LossFunction	DEFAULT_LOSS_FUNCTION_CLASS	SoftmaxCrossEntropyLoss
static final GraphWisePredictionLayerConfig[]	DEFAULT_PREDICTION_LAYER_CONFIGS	one default initialized config (See GraphWisePredictionLayerConfig)
static final EnumSet<PropertyType>	SUPPORTED_LABEL_TYPES	INTEGER, STRING, BOOLEAN, LONG
static final 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, Integer edgeEmbeddingDim, Integer seed, GraphWiseBaseConvLayerConfig[] convLayerConfigs, boolean standardize, boolean shuffle, List<String> vertexInputPropertyNames, List<String> edgeInputPropertyNames, Map<String,InputPropertyConfig> vertexInputPropertyConfigs, Map<String,InputPropertyConfig> edgeInputPropertyConfigs, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig, List<Set<String>> targetEdgeLabelSets, boolean fitted, double trainingLoss, int inputFeatureDim, int edgeInputFeatureDim, LossFunction lossFunctionClass, BatchGenerator batchGenerator, GraphWisePredictionLayerConfig[] predictionLayerConfigs, boolean normalize, 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()
Map<?,Integer>	getClassMap()
Map<?,Float>	getClassWeights()
String	getEdgeTargetPropertyName()
LabelMaps	getLabelMaps()
PropertyType	getLabelType()
LossFunction	getLossFunctionClass()
int	getNumClasses()
GraphWisePredictionLayerConfig[]	getPredictionLayerConfigs()
final void	setBatchGenerator(BatchGenerator batchGenerator)
final void	setClassMap(Map<?,Integer> classMap)
final void	setClassWeights(Map<?,Float> classWeights)
final void	setEdgeTargetPropertyName(String edgeTargetPropertyName)
final void	setLabelMaps(LabelMaps labelMaps)
final void	setLabelType(PropertyType labelType)
final void	setLossFunctionClass(LossFunction lossFunction)
final 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 Details

SUPPORTED_LABEL_TYPES

public static final EnumSet<PropertyType> SUPPORTED_LABEL_TYPES

INTEGER, STRING, BOOLEAN, LONG

SUPPORTED_REGRESSION_TYPES

public static final 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 Map<?,Float> DEFAULT_CLASS_MAP

null

Constructor Details

SupervisedEdgeWiseModelConfig

public SupervisedEdgeWiseModelConfig()

SupervisedEdgeWiseModelConfig

public SupervisedEdgeWiseModelConfig(int batchSize,
 int numEpochs,
 double learningRate,
 double weightDecay,
 int embeddingDim,
 Integer edgeEmbeddingDim,
 Integer seed,
 GraphWiseBaseConvLayerConfig[] convLayerConfigs,
 boolean standardize,
 boolean shuffle,
 List<String> vertexInputPropertyNames,
 List<String> edgeInputPropertyNames,
 Map<String,InputPropertyConfig> vertexInputPropertyConfigs,
 Map<String,InputPropertyConfig> edgeInputPropertyConfigs,
 oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig,
 List<Set<String>> targetEdgeLabelSets,
 boolean fitted,
 double trainingLoss,
 int inputFeatureDim,
 int edgeInputFeatureDim,
 LossFunction lossFunctionClass,
 BatchGenerator batchGenerator,
 GraphWisePredictionLayerConfig[] predictionLayerConfigs,
 boolean normalize,
 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 Details

getNumClasses

public int getNumClasses()

getEdgeTargetPropertyName

public String getEdgeTargetPropertyName()

setEdgeTargetPropertyName

public final void setEdgeTargetPropertyName(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 Map<?,Integer> getClassMap()

setClassMap

public final void setClassMap(Map<?,Integer> classMap)

setClassWeights

public final void setClassWeights(Map<?,Float> classWeights)

getClassWeights

public Map<?,Float> getClassWeights()

getLabelType

public PropertyType getLabelType()

setLabelType

public final void setLabelType(PropertyType labelType)

getLabelMaps

public LabelMaps getLabelMaps()

setLabelMaps

public final void setLabelMaps(LabelMaps labelMaps)