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

Modifier and Type	Class and Description
static class	oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction Deprecated.

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 and Description
static BatchGenerator	DEFAULT_BATCH_GENERATOR StandardBatchGenerator
static java.util.Map<?,java.lang.Float>	DEFAULT_CLASS_MAP null
static oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction	DEFAULT_LOSS_FUNCTION Deprecated.
static LossFunction	DEFAULT_LOSS_FUNCTION_CLASS SoftmaxCrossEntropyLoss
static boolean	DEFAULT_NORMALIZE true
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_LEARNING_RATE, DEFAULT_NUM_EPOCHS, DEFAULT_SEED, DEFAULT_SHUFFLE, DEFAULT_STANDARDIZE, DEFAULT_WEIGHT_DECAY, SUPPORTED_INPUT_TYPES

Constructor Summary

Constructors

Constructor and Description
SupervisedEdgeWiseModelConfig()
SupervisedEdgeWiseModelConfig(int batchSize, int numEpochs, double learningRate, double weightDecay, int embeddingDim, java.lang.Integer edgeEmbeddingDim, java.lang.Integer seed, GraphWiseConvLayerConfig[] convLayerConfigs, boolean standardize, boolean shuffle, java.util.List<java.lang.String> vertexInputPropertyNames, java.util.List<java.lang.String> edgeInputPropertyNames, 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)
SupervisedEdgeWiseModelConfig(SupervisedEdgeWiseModelConfig source)

Method Summary

Modifier and Type	Method and 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()
oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction	getLossFunction() Deprecated.
LossFunction	getLossFunctionClass()
int	getNumClasses()
GraphWisePredictionLayerConfig[]	getPredictionLayerConfigs()
boolean	isNormalize()
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	setLossFunction(oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction lossFunction) Deprecated.
void	setLossFunctionClass(LossFunction lossFunction)
void	setNormalize(boolean normalize)
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

getBackend, getBatchSize, getConvLayerConfigs, getEdgeInputFeatureDim, getEdgeInputPropertyNames, getEmbeddingDim, getInputFeatureDim, getLearningRate, getNumEpochs, getSeed, getTrainingLoss, getVertexInputPropertyNames, getWeightDecay, isFitted, isShuffle, isStandardize, setBatchSize, setConvLayerConfigs, setEdgeInputFeatureDim, setEdgeInputPropertyNames, setEmbeddingDim, setFitted, setInputFeatureDim, setLearningRate, setNumEpochs, setSeed, setShuffle, setStandardize, setTrainingLoss, setVertexInputPropertyNames, setWeightDecay

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_BATCH_GENERATOR

public static final BatchGenerator DEFAULT_BATCH_GENERATOR

StandardBatchGenerator

DEFAULT_CLASS_MAP

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

null

DEFAULT_LOSS_FUNCTION

@Deprecated
public static final oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction DEFAULT_LOSS_FUNCTION

Deprecated.

SupervisedEdgeWiseModelConfig.LossFunction.SOFTMAX_CROSS_ENTROPY

DEFAULT_LOSS_FUNCTION_CLASS

public static final LossFunction DEFAULT_LOSS_FUNCTION_CLASS

SoftmaxCrossEntropyLoss

DEFAULT_NORMALIZE

public static final boolean DEFAULT_NORMALIZE

true

See Also:

Constant Field Values

DEFAULT_PREDICTION_LAYER_CONFIGS

public static final GraphWisePredictionLayerConfig[] DEFAULT_PREDICTION_LAYER_CONFIGS

one default initialized config (See GraphWisePredictionLayerConfig)

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

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,
                                     GraphWiseConvLayerConfig[] convLayerConfigs,
                                     boolean standardize,
                                     boolean shuffle,
                                     java.util.List<java.lang.String> vertexInputPropertyNames,
                                     java.util.List<java.lang.String> edgeInputPropertyNames,
                                     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)

SupervisedEdgeWiseModelConfig

public SupervisedEdgeWiseModelConfig(SupervisedEdgeWiseModelConfig source)

Method Detail

getBatchGenerator

public BatchGenerator getBatchGenerator()

getClassMap

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

getClassWeights

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

getEdgeTargetPropertyName

public java.lang.String getEdgeTargetPropertyName()

getLabelMaps

public LabelMaps getLabelMaps()

getLabelType

public PropertyType getLabelType()

getLossFunction

@Deprecated
public oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction getLossFunction()

Deprecated.

getLossFunctionClass

public LossFunction getLossFunctionClass()

getNumClasses

public int getNumClasses()

getPredictionLayerConfigs

public GraphWisePredictionLayerConfig[] getPredictionLayerConfigs()

isNormalize

public boolean isNormalize()

setBatchGenerator

public final void setBatchGenerator(BatchGenerator batchGenerator)

setClassMap

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

setClassWeights

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

setEdgeTargetPropertyName

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

setLabelMaps

public final void setLabelMaps(LabelMaps labelMaps)

setLabelType

public final void setLabelType(PropertyType labelType)

setLossFunction

@Deprecated
public final void setLossFunction(oracle.pgx.config.mllib.SupervisedEdgeWiseModelConfig.LossFunction lossFunction)

Deprecated.

setLossFunctionClass

public final void setLossFunctionClass(LossFunction lossFunction)

setNormalize

public final void setNormalize(boolean normalize)

setPredictionLayerConfigs

public final void setPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs)