|
NeuralNet 1.0
|
#include <SGD.hpp>
Public Member Functions | |
| SGD (double alpha) | |
| void | updateWeights (Eigen::MatrixXd &weights, const Eigen::MatrixXd &weightsGrad) override |
| This function updates the weights passed based on the selected Optimizer and the weights gradients. | |
| void | updateBiases (Eigen::MatrixXd &biases, const Eigen::MatrixXd &biasesGrad) override |
| This function updates the biases passed based based on the Optimizer and the biases gradients. | |
Public Member Functions inherited from NeuralNet::Optimizer | |
| Optimizer (double alpha) | |
Additional Inherited Members | |
Protected Attributes inherited from NeuralNet::Optimizer | |
| double | alpha |
Stochastic Gradient Descent optimizer
|
inlineoverridevirtual |
This function updates the biases passed based based on the Optimizer and the biases gradients.
| biases | The biases that should be updated |
| biasesGrad | The biases gradient |
The function will return void, since it only performs an update on the biases passed
Implements NeuralNet::Optimizer.
|
inlineoverridevirtual |
This function updates the weights passed based on the selected Optimizer and the weights gradients.
| weights | The weights that should be updated |
| weightsGrad | The weights gradient |
The function will return void, since it only performs an update on the weights passed
Implements NeuralNet::Optimizer.