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. | |
![]() | |
Optimizer (double alpha) | |
Additional Inherited Members | |
![]() | |
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.