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 GraphWiseEmbeddingConfigpublic void setAlpha(double alpha)
alpha - the feature reconstruction weight. Must be between 0 and 1public final void setDecoderLayerConfigs(GraphWisePredictionLayerConfig... decoderLayerConfigs)
decoderLayerConfigs - the configuration objects