public abstract class GraphWiseModelConfig
extends java.lang.Object
Modifier and Type | Class and Description |
---|---|
static class |
GraphWiseModelConfig.Backend |
static class |
GraphWiseModelConfig.GraphConvModelVariant |
Modifier and Type | Field and Description |
---|---|
static GraphWiseModelConfig.Backend |
DEFAULT_BACKEND
LibTorch
|
static int |
DEFAULT_BATCH_SIZE
128
|
static GraphWiseConvLayerConfig[] |
DEFAULT_CONV_LAYER_CONFIGS
two default initialized layer configs (See
GraphWiseConvLayerConfig ) |
static int |
DEFAULT_EMBEDDING_DIM
128
|
static double |
DEFAULT_LEARNING_RATE
0.01
|
static GraphWiseModelConfig.GraphConvModelVariant |
DEFAULT_MODE
null
|
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 |
---|
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, GraphWiseModelConfig.Backend backend, GraphWiseModelConfig.GraphConvModelVariant variant) |
Modifier and Type | Method and Description |
---|---|
GraphWiseModelConfig.Backend |
getBackend() |
int |
getBatchSize() |
GraphWiseConvLayerConfig[] |
getConvLayerConfigs() |
int |
getEdgeInputFeatureDim() |
java.util.List<java.lang.String> |
getEdgeInputPropertyNames() |
int |
getEmbeddingDim() |
int |
getInputFeatureDim() |
double |
getLearningRate() |
int |
getNumEpochs() |
java.lang.Integer |
getSeed() |
java.util.List<java.util.Set<java.lang.String>> |
getTargetVertexLabelSets() |
double |
getTrainingLoss() |
GraphWiseModelConfig.GraphConvModelVariant |
getVariant() |
java.util.List<java.lang.String> |
getVertexInputPropertyNames() |
double |
getWeightDecay() |
boolean |
isFitted() |
boolean |
isShuffle() |
boolean |
isStandardize() |
void |
setBatchSize(int batchSize) |
void |
setConvLayerConfigs(GraphWiseConvLayerConfig... convLayerConfigs) |
void |
setEdgeInputFeatureDim(int edgeInputFeatureDim) |
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 |
setNumEpochs(int numEpochs) |
void |
setSeed(int seed) |
void |
setShuffle(boolean shuffle) |
void |
setStandardize(boolean standardize) |
void |
setTargetVertexLabels(java.util.List<java.lang.String> targetVertexLabels) |
void |
setTargetVertexLabelSets(java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets) |
void |
setTrainingLoss(double trainingLoss) |
void |
setVariant(GraphWiseModelConfig.GraphConvModelVariant variant) |
void |
setVertexInputPropertyNames(java.util.List<java.lang.String> vertexInputPropertyNames) |
void |
setWeightDecay(double weightDecay) |
public static final GraphWiseModelConfig.Backend DEFAULT_BACKEND
public static final int DEFAULT_BATCH_SIZE
public static final GraphWiseConvLayerConfig[] DEFAULT_CONV_LAYER_CONFIGS
GraphWiseConvLayerConfig
)public static final int DEFAULT_EMBEDDING_DIM
public static final double DEFAULT_LEARNING_RATE
public static final GraphWiseModelConfig.GraphConvModelVariant DEFAULT_MODE
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 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, GraphWiseModelConfig.Backend backend, GraphWiseModelConfig.GraphConvModelVariant variant)
public GraphWiseModelConfig.Backend getBackend()
public int getBatchSize()
public GraphWiseConvLayerConfig[] getConvLayerConfigs()
public int getEdgeInputFeatureDim()
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 java.util.List<java.util.Set<java.lang.String>> getTargetVertexLabelSets()
public double getTrainingLoss()
public GraphWiseModelConfig.GraphConvModelVariant getVariant()
public java.util.List<java.lang.String> getVertexInputPropertyNames()
public double getWeightDecay()
public boolean isFitted()
public boolean isShuffle()
public boolean isStandardize()
public final void setBatchSize(int batchSize)
public final void setConvLayerConfigs(GraphWiseConvLayerConfig... convLayerConfigs)
public final void setEdgeInputFeatureDim(int edgeInputFeatureDim)
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 setNumEpochs(int numEpochs)
public final void setSeed(int seed)
public final void setShuffle(boolean shuffle)
public final void setStandardize(boolean standardize)
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 setTrainingLoss(double trainingLoss)
public final void setVariant(GraphWiseModelConfig.GraphConvModelVariant variant)
public final void setVertexInputPropertyNames(java.util.List<java.lang.String> vertexInputPropertyNames)
public final void setWeightDecay(double weightDecay)