Class that specifies parameters of the neural network layer.
More...
◆ Parameter()
Constructs the parameter of the neural network layer
- Parameters
-
context | Context to manage the parameter of the neural network layer |
◆ getPredictionStage()
boolean getPredictionStage |
( |
| ) |
|
Gets the flag specifying whether the layer is used for the prediction stage or not
◆ getWeightsAndBiasesInitializationFlag()
boolean getWeightsAndBiasesInitializationFlag |
( |
| ) |
|
Gets the flag that specifies whether the weights and biases are initialized or not.
◆ setBiasesInitializer()
Sets the layer biases initializer
- Parameters
-
biasesInitializer | Layer biases initializer |
◆ setPredictionStage()
void setPredictionStage |
( |
boolean |
predictionStage | ) |
|
Sets the flag specifying whether the layer is used for the prediction stage or not
- Parameters
-
predictionStage | Flag specifying whether the layer is used for the prediction stage or not |
◆ setWeightsAndBiasesInitializationFlag()
void setWeightsAndBiasesInitializationFlag |
( |
boolean |
weightsAndBiasesInitialized | ) |
|
Sets the flag that specifies whether the weights and biases are initialized or not.
- Parameters
-
weightsAndBiasesInitialized | Flag that specifies whether the weights and biases are initialized or not. |
◆ setWeightsInitializer()
Sets the layer weights initializer
- Parameters
-
weightsInitializer | Layer weights initializer |
The documentation for this class was generated from the following file:
- neural_networks/layers/Parameter.java