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 class	EdgeWiseModelConfig.EdgeWiseConvModelVariant

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

GraphWiseBaseModelConfig.Backend

Field Summary

Fields

Modifier and Type	Field	Description
static EdgeWiseModelConfig.EdgeWiseConvModelVariant	DEFAULT_MODE	null
static java.lang.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, 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, java.util.List<java.util.Set<java.lang.String>> targetEdgeLabelSets, GraphWiseBaseModelConfig.Backend backend, java.lang.Integer edgeEmbeddingDim, EdgeWiseModelConfig.EdgeWiseConvModelVariant variant, EdgeCombinationMethod edgeCombinationMethod, boolean enableAccelerator, GraphWiseValidationConfig validationConfig)

Method Summary

Modifier and Type	Method	Description
EdgeCombinationMethod	getEdgeCombinationMethod()
java.lang.Integer	getEdgeEmbeddingDim()
java.util.List<java.util.Set<java.lang.String>>	getTargetEdgeLabelSets()
EdgeWiseModelConfig.EdgeWiseConvModelVariant	getVariant()
void	setEdgeCombinationMethod(EdgeCombinationMethod edgeCombinationMethod)
void	setEdgeEmbeddingDim(java.lang.Integer edgeEmbeddingDim)
void	setTargetEdgeLabels(java.util.List<java.lang.String> targetEdgeLabels)
void	setTargetEdgeLabelSets(java.util.List<java.util.Set<java.lang.String>> targetEdgeLabelSets)
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 Detail

EDGE_EMBEDDING_SIZE

public static final java.lang.Integer EDGE_EMBEDDING_SIZE

null

DEFAULT_MODE

public static final EdgeWiseModelConfig.EdgeWiseConvModelVariant DEFAULT_MODE

null

Constructor Detail

EdgeWiseModelConfig

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

Method Detail

getTargetEdgeLabelSets

public java.util.List<java.util.Set<java.lang.String>> getTargetEdgeLabelSets()

setTargetEdgeLabelSets

public final void setTargetEdgeLabelSets(java.util.List<java.util.Set<java.lang.String>> targetEdgeLabelSets)

setTargetEdgeLabels

public void setTargetEdgeLabels(java.util.List<java.lang.String> targetEdgeLabels)

getEdgeEmbeddingDim

public java.lang.Integer getEdgeEmbeddingDim()

setEdgeEmbeddingDim

public void setEdgeEmbeddingDim(java.lang.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()