public class GraphWisePredictionLayerConfig
extends java.lang.Object
GraphWisePredictionLayerConfigBuilder
for a description of the hyperparameters.Modifier and Type | Field and Description |
---|---|
static ActivationFunction |
DEFAULT_ACTIVATION_FUNCTION
|
static double |
DEFAULT_DROPOUT_RATE
0.
|
static java.lang.Integer |
DEFAULT_HIDDEN_DIMENSION
null
|
static WeightInitScheme |
DEFAULT_WEIGHT_INIT_SCHEME
|
Constructor and Description |
---|
GraphWisePredictionLayerConfig() |
Modifier and Type | Method and Description |
---|---|
ActivationFunction |
getActivationFunction() |
double |
getDropoutRate() |
java.lang.Integer |
getHiddenDimension() |
WeightInitScheme |
getWeightInitScheme() |
void |
setActivationFunction(ActivationFunction activationFunction) |
void |
setDropoutRate(double dropoutRate) |
void |
setHiddenDimension(java.lang.Integer hiddenDimension) |
void |
setWeightInitScheme(WeightInitScheme weightInitScheme) |
public static final ActivationFunction DEFAULT_ACTIVATION_FUNCTION
public static final double DEFAULT_DROPOUT_RATE
public static final java.lang.Integer DEFAULT_HIDDEN_DIMENSION
public static final WeightInitScheme DEFAULT_WEIGHT_INIT_SCHEME
public ActivationFunction getActivationFunction()
public double getDropoutRate()
public java.lang.Integer getHiddenDimension()
public WeightInitScheme getWeightInitScheme()
public void setActivationFunction(ActivationFunction activationFunction)
public void setDropoutRate(double dropoutRate)
public void setHiddenDimension(java.lang.Integer hiddenDimension)
public void setWeightInitScheme(WeightInitScheme weightInitScheme)