public abstract class GraphWiseModelConfig extends GraphWiseBaseModelConfig
Modifier and Type | Class and Description |
---|---|
static class |
GraphWiseModelConfig.GraphConvModelVariant |
GraphWiseBaseModelConfig.Backend
Modifier and Type | Field and Description |
---|---|
static GraphWiseModelConfig.GraphConvModelVariant |
DEFAULT_MODE
null
|
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 |
---|
GraphWiseModelConfig(GraphWiseModelConfig source,
oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig) |
GraphWiseModelConfig(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>> targetVertexLabelSets,
GraphWiseBaseModelConfig.Backend backend,
GraphWiseModelConfig.GraphConvModelVariant variant) |
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) |
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 GraphWiseModelConfig.GraphConvModelVariant DEFAULT_MODE
public GraphWiseModelConfig(GraphWiseModelConfig source, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
public GraphWiseModelConfig(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>> targetVertexLabelSets, GraphWiseBaseModelConfig.Backend backend, GraphWiseModelConfig.GraphConvModelVariant variant)
public java.util.List<java.util.Set<java.lang.String>> getTargetVertexLabelSets()
public GraphWiseModelConfig.GraphConvModelVariant getVariant()
public void setTargetVertexLabels(java.util.List<java.lang.String> targetVertexLabels)
public final void setTargetVertexLabelSets(java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets)
public final void setVariant(GraphWiseModelConfig.GraphConvModelVariant variant)