public abstract class GraphWiseBaseModelConfig
extends java.lang.Object
Modifier and Type | Class and Description |
---|---|
static class |
GraphWiseBaseModelConfig.Backend |
Modifier and Type | Field and Description |
---|---|
static GraphWiseBaseModelConfig.Backend |
DEFAULT_BACKEND
LibTorch
|
static int |
DEFAULT_BATCH_SIZE
128
|
static GraphWiseBaseConvLayerConfig[] |
DEFAULT_CONV_LAYER_CONFIGS
two default initialized layer configs (See
GraphWiseBaseConvLayerConfig ) |
static int |
DEFAULT_EMBEDDING_DIM
128
|
static double |
DEFAULT_LEARNING_RATE
0.01
|
static boolean |
DEFAULT_NORMALIZE
true
|
static int |
DEFAULT_NUM_EPOCHS
3
|
static java.lang.Integer |
DEFAULT_SEED
null
|
static boolean |
DEFAULT_SHUFFLE
true
|
static boolean |
DEFAULT_STANDARDIZE
false
|
static double |
DEFAULT_WEIGHT_DECAY
0.0
|
static java.util.EnumSet<PropertyType> |
SUPPORTED_INPUT_TYPES
BOOLEAN, FLOAT, DOUBLE
|
Constructor and Description |
---|
GraphWiseBaseModelConfig(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,
GraphWiseBaseModelConfig.Backend backend) |
Modifier and Type | Method and Description |
---|---|
void |
clearSensitiveData() |
GraphWiseBaseModelConfig.Backend |
getBackend() |
int |
getBatchSize() |
oracle.pgx.config.internal.categorymapping.CategoryMappingConfig |
getCategoryMappingConfig() |
GraphWiseBaseConvLayerConfig[] |
getConvLayerConfigs() |
int |
getEdgeInputFeatureDim() |
java.util.Map<java.lang.String,InputPropertyConfig> |
getEdgeInputPropertyConfigs() |
java.util.List<java.lang.String> |
getEdgeInputPropertyNames() |
int |
getEmbeddingDim() |
int |
getInputFeatureDim() |
double |
getLearningRate() |
int |
getNumEpochs() |
java.lang.Integer |
getSeed() |
double |
getTrainingLoss() |
java.util.Map<java.lang.String,InputPropertyConfig> |
getVertexInputPropertyConfigs() |
java.util.List<java.lang.String> |
getVertexInputPropertyNames() |
double |
getWeightDecay() |
boolean |
isFitted() |
boolean |
isNormalize() |
boolean |
isShuffle() |
boolean |
isStandardize() |
void |
setBatchSize(int batchSize) |
void |
setCategoryMappingConfig(oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig) |
void |
setConvLayerConfigs(GraphWiseBaseConvLayerConfig... convLayerConfigs) |
void |
setEdgeInputFeatureDim(int edgeInputFeatureDim) |
void |
setEdgeInputPropertyConfigs(InputPropertyConfig... edgeInputPropertyConfigs) |
void |
setEdgeInputPropertyNames(java.util.List<java.lang.String> edgeInputPropertyNames) |
void |
setEmbeddingDim(int embeddingDim) |
void |
setFitted(boolean fitted) |
void |
setInputFeatureDim(int vertexInputFeatureDim) |
void |
setLearningRate(double learningRate) |
void |
setNormalize(boolean normalize) |
void |
setNumEpochs(int numEpochs) |
void |
setSeed(int seed) |
void |
setShuffle(boolean shuffle) |
void |
setStandardize(boolean standardize) |
void |
setTrainingLoss(double trainingLoss) |
void |
setVertexInputPropertyConfigs(InputPropertyConfig... vertexInputPropertyConfigs) |
void |
setVertexInputPropertyNames(java.util.List<java.lang.String> vertexInputPropertyNames) |
void |
setWeightDecay(double weightDecay) |
public static final GraphWiseBaseModelConfig.Backend DEFAULT_BACKEND
public static final int DEFAULT_BATCH_SIZE
public static final GraphWiseBaseConvLayerConfig[] DEFAULT_CONV_LAYER_CONFIGS
GraphWiseBaseConvLayerConfig
)public static final int DEFAULT_EMBEDDING_DIM
public static final double DEFAULT_LEARNING_RATE
public static final boolean DEFAULT_NORMALIZE
public static final int DEFAULT_NUM_EPOCHS
public static final java.lang.Integer DEFAULT_SEED
public static final boolean DEFAULT_SHUFFLE
public static final boolean DEFAULT_STANDARDIZE
public static final double DEFAULT_WEIGHT_DECAY
public static final java.util.EnumSet<PropertyType> SUPPORTED_INPUT_TYPES
public GraphWiseBaseModelConfig(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, GraphWiseBaseModelConfig.Backend backend)
public final void clearSensitiveData()
public GraphWiseBaseModelConfig.Backend getBackend()
public int getBatchSize()
public oracle.pgx.config.internal.categorymapping.CategoryMappingConfig getCategoryMappingConfig()
public GraphWiseBaseConvLayerConfig[] getConvLayerConfigs()
public int getEdgeInputFeatureDim()
public java.util.Map<java.lang.String,InputPropertyConfig> getEdgeInputPropertyConfigs()
public java.util.List<java.lang.String> getEdgeInputPropertyNames()
public int getEmbeddingDim()
public int getInputFeatureDim()
public double getLearningRate()
public int getNumEpochs()
public java.lang.Integer getSeed()
public double getTrainingLoss()
public java.util.Map<java.lang.String,InputPropertyConfig> getVertexInputPropertyConfigs()
public java.util.List<java.lang.String> getVertexInputPropertyNames()
public double getWeightDecay()
public boolean isFitted()
public boolean isNormalize()
public boolean isShuffle()
public boolean isStandardize()
public final void setBatchSize(int batchSize)
public final void setCategoryMappingConfig(oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
public final void setConvLayerConfigs(GraphWiseBaseConvLayerConfig... convLayerConfigs)
public final void setEdgeInputFeatureDim(int edgeInputFeatureDim)
public final void setEdgeInputPropertyConfigs(InputPropertyConfig... edgeInputPropertyConfigs)
public final void setEdgeInputPropertyNames(java.util.List<java.lang.String> edgeInputPropertyNames)
public final void setEmbeddingDim(int embeddingDim)
public final void setFitted(boolean fitted)
public final void setInputFeatureDim(int vertexInputFeatureDim)
public final void setLearningRate(double learningRate)
public final void setNormalize(boolean normalize)
public final void setNumEpochs(int numEpochs)
public final void setSeed(int seed)
public final void setShuffle(boolean shuffle)
public final void setStandardize(boolean standardize)
public final void setTrainingLoss(double trainingLoss)
public final void setVertexInputPropertyConfigs(InputPropertyConfig... vertexInputPropertyConfigs)
public final void setVertexInputPropertyNames(java.util.List<java.lang.String> vertexInputPropertyNames)
public final void setWeightDecay(double weightDecay)