oracle.pgx.config.mllib

Class GraphWiseModelConfig

java.lang.Object

oracle.pgx.config.mllib.GraphWiseBaseModelConfig

oracle.pgx.config.mllib.GraphWiseModelConfig

Direct Known Subclasses:

SupervisedGraphWiseModelConfig, UnsupervisedGraphWiseModelConfig

public abstract class GraphWiseModelConfig
extends GraphWiseBaseModelConfig

Abstract configuration class for GraphWise models

Since:

19.4

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	GraphWiseModelConfig.GraphConvModelVariant

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

GraphWiseBaseModelConfig.Backend

Field Summary

Fields

Modifier and Type	Field and Description
static GraphWiseModelConfig.GraphConvModelVariant	DEFAULT_MODE 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

GraphWiseModelConfig(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>> targetVertexLabelSets, GraphWiseBaseModelConfig.Backend backend, GraphWiseModelConfig.GraphConvModelVariant variant)

Method Summary

Modifier and Type	Method and Description
java.util.List<java.util.Set<java.lang.String>>	getTargetVertexLabelSets()
GraphWiseModelConfig.GraphConvModelVariant	getVariant()
void	setTargetVertexLabels(java.util.List<java.lang.String> targetVertexLabels)
void	setTargetVertexLabelSets(java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets)
void	setVariant(GraphWiseModelConfig.GraphConvModelVariant 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 GraphWiseModelConfig.GraphConvModelVariant DEFAULT_MODE

null

Constructor Detail

GraphWiseModelConfig

public GraphWiseModelConfig(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>> targetVertexLabelSets,
                            GraphWiseBaseModelConfig.Backend backend,
                            GraphWiseModelConfig.GraphConvModelVariant variant)

Method Detail

getTargetVertexLabelSets

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

getVariant

public GraphWiseModelConfig.GraphConvModelVariant getVariant()

setTargetVertexLabels

public void setTargetVertexLabels(java.util.List<java.lang.String> targetVertexLabels)

setTargetVertexLabelSets

public final void setTargetVertexLabelSets(java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets)

setVariant

public final void setVariant(GraphWiseModelConfig.GraphConvModelVariant variant)