oracle.pgx.config.mllib

Class EdgeWiseModelConfig

java.lang.Object

oracle.pgx.config.mllib.GraphWiseBaseModelConfig

oracle.pgx.config.mllib.EdgeWiseModelConfig

Direct Known Subclasses:

SupervisedEdgeWiseModelConfig

public abstract class EdgeWiseModelConfig
extends GraphWiseBaseModelConfig

Abstract configuration class for EdgeWise models

Since:

23.1

Nested Class Summary

Nested Classes

Modifier and Type	Class and 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 and 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_LEARNING_RATE, DEFAULT_NUM_EPOCHS, DEFAULT_SEED, DEFAULT_SHUFFLE, DEFAULT_STANDARDIZE, DEFAULT_WEIGHT_DECAY, SUPPORTED_INPUT_TYPES

Constructor Summary

Constructors

Constructor and Description

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

Method Summary

Modifier and Type	Method and 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

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_MODE

public static final EdgeWiseModelConfig.EdgeWiseConvModelVariant DEFAULT_MODE

null

EDGE_EMBEDDING_SIZE

public static final java.lang.Integer EDGE_EMBEDDING_SIZE

null

Constructor Detail

EdgeWiseModelConfig

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

Method Detail

getEdgeCombinationMethod

public EdgeCombinationMethod getEdgeCombinationMethod()

getEdgeEmbeddingDim

public java.lang.Integer getEdgeEmbeddingDim()

getTargetEdgeLabelSets

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

getVariant

public EdgeWiseModelConfig.EdgeWiseConvModelVariant getVariant()

setEdgeCombinationMethod

public void setEdgeCombinationMethod(EdgeCombinationMethod edgeCombinationMethod)

setEdgeEmbeddingDim

public void setEdgeEmbeddingDim(java.lang.Integer edgeEmbeddingDim)

setTargetEdgeLabels

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

setTargetEdgeLabelSets

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

setVariant

public final void setVariant(EdgeWiseModelConfig.EdgeWiseConvModelVariant variant)