public class UnsupervisedEdgeWiseModelConfig extends EdgeWiseModelConfig
UnsupervisedEdgeWiseModel
.
See UnsupervisedEdgeWiseModel
for a description of the hyperparameters.Modifier and Type | Class and Description |
---|---|
static class |
UnsupervisedEdgeWiseModelConfig.LossFunction |
EdgeWiseModelConfig.EdgeWiseConvModelVariant
GraphWiseBaseModelConfig.Backend
Modifier and Type | Field and Description |
---|---|
static GraphWiseDgiLayerConfig |
DEFAULT_DGI_LAYER_CONFIG
one default initialized config (See
GraphWiseDgiLayerConfig ) |
static UnsupervisedEdgeWiseModelConfig.LossFunction |
DEFAULT_LOSS_FUNCTION
|
DEFAULT_MODE, EDGE_EMBEDDING_SIZE
DEFAULT_BACKEND, DEFAULT_BATCH_SIZE, DEFAULT_CONV_LAYER_CONFIGS, DEFAULT_EMBEDDING_DIM, DEFAULT_LEARNING_RATE, DEFAULT_NORMALIZE, DEFAULT_NUM_EPOCHS, DEFAULT_SEED, DEFAULT_SHUFFLE, DEFAULT_STANDARDIZE, DEFAULT_WEIGHT_DECAY, SUPPORTED_INPUT_TYPES
Constructor and Description |
---|
UnsupervisedEdgeWiseModelConfig() |
UnsupervisedEdgeWiseModelConfig(int batchSize,
int numEpochs,
double learningRate,
double weightDecay,
int embeddingDim,
java.lang.Integer edgeEmbeddingDim,
java.lang.Integer seed,
GraphWiseBaseConvLayerConfig[] convLayerConfigs,
boolean standardize,
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,
java.util.List<java.util.Set<java.lang.String>> targetEdgeLabelSets,
boolean fitted,
double trainingLoss,
int inputFeatureDim,
int edgeInputFeatureDim,
UnsupervisedEdgeWiseModelConfig.LossFunction lossFunction,
boolean normalize,
GraphWiseBaseModelConfig.Backend backend,
EdgeCombinationMethod edgeCombinationMethod,
EdgeWiseModelConfig.EdgeWiseConvModelVariant variant,
GraphWiseDgiLayerConfig dgiLayerConfig) |
UnsupervisedEdgeWiseModelConfig(UnsupervisedEdgeWiseModelConfig source) |
UnsupervisedEdgeWiseModelConfig(UnsupervisedEdgeWiseModelConfig source,
oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig) |
Modifier and Type | Method and Description |
---|---|
GraphWiseDgiLayerConfig |
getDgiLayerConfig() |
UnsupervisedEdgeWiseModelConfig.LossFunction |
getLossFunction() |
void |
setDgiLayerConfig(GraphWiseDgiLayerConfig dgiLayerConfig) |
void |
setLossFunction(UnsupervisedEdgeWiseModelConfig.LossFunction lossFunction) |
getEdgeCombinationMethod, getEdgeEmbeddingDim, getTargetEdgeLabelSets, getVariant, setEdgeCombinationMethod, setEdgeEmbeddingDim, setTargetEdgeLabels, setTargetEdgeLabelSets, setVariant
clearSensitiveData, getBackend, getBatchSize, getCategoryMappingConfig, getConvLayerConfigs, getEdgeInputFeatureDim, getEdgeInputPropertyConfigs, getEdgeInputPropertyNames, getEmbeddingDim, getInputFeatureDim, getLearningRate, getNumEpochs, getSeed, getTrainingLoss, getVertexInputPropertyConfigs, getVertexInputPropertyNames, getWeightDecay, isFitted, isNormalize, isShuffle, isStandardize, setBatchSize, setCategoryMappingConfig, setConvLayerConfigs, setEdgeInputFeatureDim, setEdgeInputPropertyConfigs, setEdgeInputPropertyNames, setEmbeddingDim, setFitted, setInputFeatureDim, setLearningRate, setNormalize, setNumEpochs, setSeed, setShuffle, setStandardize, setTrainingLoss, setVertexInputPropertyConfigs, setVertexInputPropertyNames, setWeightDecay
public static final GraphWiseDgiLayerConfig DEFAULT_DGI_LAYER_CONFIG
GraphWiseDgiLayerConfig
)public static final UnsupervisedEdgeWiseModelConfig.LossFunction DEFAULT_LOSS_FUNCTION
public UnsupervisedEdgeWiseModelConfig()
public UnsupervisedEdgeWiseModelConfig(int batchSize, int numEpochs, double learningRate, double weightDecay, int embeddingDim, java.lang.Integer edgeEmbeddingDim, java.lang.Integer seed, GraphWiseBaseConvLayerConfig[] convLayerConfigs, boolean standardize, 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, java.util.List<java.util.Set<java.lang.String>> targetEdgeLabelSets, boolean fitted, double trainingLoss, int inputFeatureDim, int edgeInputFeatureDim, UnsupervisedEdgeWiseModelConfig.LossFunction lossFunction, boolean normalize, GraphWiseBaseModelConfig.Backend backend, EdgeCombinationMethod edgeCombinationMethod, EdgeWiseModelConfig.EdgeWiseConvModelVariant variant, GraphWiseDgiLayerConfig dgiLayerConfig)
public UnsupervisedEdgeWiseModelConfig(UnsupervisedEdgeWiseModelConfig source)
public UnsupervisedEdgeWiseModelConfig(UnsupervisedEdgeWiseModelConfig source, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
public GraphWiseDgiLayerConfig getDgiLayerConfig()
public UnsupervisedEdgeWiseModelConfig.LossFunction getLossFunction()
public final void setDgiLayerConfig(GraphWiseDgiLayerConfig dgiLayerConfig)
public final void setLossFunction(UnsupervisedEdgeWiseModelConfig.LossFunction lossFunction)