public class GraphWiseDominantLayerConfig extends java.lang.Object implements GraphWiseEmbeddingConfig
GraphWiseDominantLayerConfigBuilder
for a description of the
hyperparameters.GraphWiseEmbeddingConfig.EmbeddingType
Modifier and Type | Field and Description |
---|---|
static double |
DEFAULT_ALPHA
Default alpha parameter as 0.5: equal weight for feature and structure reconstruction losses
|
static GraphWisePredictionLayerConfig[] |
DEFAULT_PREDICTION_LAYER_CONFIGS
by default no prediction layer config
|
Constructor and Description |
---|
GraphWiseDominantLayerConfig() |
Modifier and Type | Method and Description |
---|---|
double |
getAlpha() |
GraphWisePredictionLayerConfig[] |
getdecoderLayerConfigs()
Gets the configuration objects for the decoder layers
|
GraphWiseEmbeddingConfig.EmbeddingType |
getEmbeddingType() |
void |
setAlpha(double alpha)
Sets the weight for feature reconstruction loss.
|
void |
setDecoderLayerConfigs(GraphWisePredictionLayerConfig... decoderLayerConfigs)
Sets the configuration objects for the decoder layers
|
public static final double DEFAULT_ALPHA
public static final GraphWisePredictionLayerConfig[] DEFAULT_PREDICTION_LAYER_CONFIGS
public double getAlpha()
public GraphWisePredictionLayerConfig[] getdecoderLayerConfigs()
public GraphWiseEmbeddingConfig.EmbeddingType getEmbeddingType()
getEmbeddingType
in interface GraphWiseEmbeddingConfig
public void setAlpha(double alpha)
alpha
- the feature reconstruction weight. Must be between 0 and 1public final void setDecoderLayerConfigs(GraphWisePredictionLayerConfig... decoderLayerConfigs)
decoderLayerConfigs
- the configuration objects