Package oracle.pgx.config.mllib

Class EdgeWiseModelConfig

java.lang.Object

oracle.pgx.config.mllib.GraphWiseBaseModelConfig

oracle.pgx.config.mllib.EdgeWiseModelConfig

Direct Known Subclasses:

SupervisedEdgeWiseModelConfig, UnsupervisedEdgeWiseModelConfig

public abstract class EdgeWiseModelConfig
extends GraphWiseBaseModelConfig

Abstract configuration class for EdgeWise models

Since:

23.1

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	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 EdgeWiseModelConfig.EdgeWiseConvModelVariant	DEFAULT_MODE	null
static final Integer	EDGE_EMBEDDING_SIZE	null

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

EdgeWiseModelConfig(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, List<Set<String>> targetEdgeLabelSets, GraphWiseBaseModelConfig.Backend backend, Integer edgeEmbeddingDim, EdgeWiseModelConfig.EdgeWiseConvModelVariant variant, EdgeCombinationMethod edgeCombinationMethod, boolean enableAccelerator, GraphWiseValidationConfig validationConfig)

Method Summary

Modifier and Type	Method	Description
EdgeCombinationMethod	getEdgeCombinationMethod()
Integer	getEdgeEmbeddingDim()
List<Set<String>>	getTargetEdgeLabelSets()
EdgeWiseModelConfig.EdgeWiseConvModelVariant	getVariant()
void	setEdgeCombinationMethod(EdgeCombinationMethod edgeCombinationMethod)
void	setEdgeEmbeddingDim(Integer edgeEmbeddingDim)
void	setTargetEdgeLabels(List<String> targetEdgeLabels)
final void	setTargetEdgeLabelSets(List<Set<String>> targetEdgeLabelSets)
final void	setVariant(EdgeWiseModelConfig.EdgeWiseConvModelVariant variant)

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

EDGE_EMBEDDING_SIZE

public static final Integer EDGE_EMBEDDING_SIZE

null

DEFAULT_MODE

public static final EdgeWiseModelConfig.EdgeWiseConvModelVariant DEFAULT_MODE

null

Constructor Details

EdgeWiseModelConfig

public EdgeWiseModelConfig(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,
 List<Set<String>> targetEdgeLabelSets,
 GraphWiseBaseModelConfig.Backend backend,
 Integer edgeEmbeddingDim,
 EdgeWiseModelConfig.EdgeWiseConvModelVariant variant,
 EdgeCombinationMethod edgeCombinationMethod,
 boolean enableAccelerator,
 GraphWiseValidationConfig validationConfig)

Method Details

getTargetEdgeLabelSets

public List<Set<String>> getTargetEdgeLabelSets()

setTargetEdgeLabelSets

public final void setTargetEdgeLabelSets(List<Set<String>> targetEdgeLabelSets)

setTargetEdgeLabels

public void setTargetEdgeLabels(List<String> targetEdgeLabels)

getEdgeEmbeddingDim

public Integer getEdgeEmbeddingDim()

setEdgeEmbeddingDim

public void setEdgeEmbeddingDim(Integer edgeEmbeddingDim)

getEdgeCombinationMethod

public EdgeCombinationMethod getEdgeCombinationMethod()

setEdgeCombinationMethod

public void setEdgeCombinationMethod(EdgeCombinationMethod edgeCombinationMethod)

setVariant

public final void setVariant(EdgeWiseModelConfig.EdgeWiseConvModelVariant variant)

getVariant

public EdgeWiseModelConfig.EdgeWiseConvModelVariant getVariant()