NeuralNet/cpp-docs/Optimizer_8hpp_source.html

#pragma once


#include <Eigen/Dense>


namespace NeuralNet {


class Optimizer {

  friend class Network;


 public:

  Optimizer(double alpha) : alpha(alpha){};


  virtual ~Optimizer() = default;  // Virtual destructor


  virtual void updateWeights(Eigen::MatrixXd &weights,

                             const Eigen::MatrixXd &weightsGrad) = 0;


  virtual void updateBiases(Eigen::MatrixXd &biases,

                            const Eigen::MatrixXd &biasesGrad) = 0;


 protected:

  double alpha;


  virtual void insiderInit(size_t size) = 0;

};


}  // namespace NeuralNet

NeuralNet::Network
Definition Network.hpp:29

NeuralNet::Optimizer
Definition Optimizer.hpp:6

NeuralNet::Optimizer::updateBiases
virtual void updateBiases(Eigen::MatrixXd &biases, const Eigen::MatrixXd &biasesGrad)=0
This function updates the biases passed based based on the Optimizer and the biases gradients.

NeuralNet::Optimizer::insiderInit
virtual void insiderInit(size_t size)=0
This function's purpose is to provide an interface to perform updates for the Optimizers from within ...

NeuralNet::Optimizer::updateWeights
virtual void updateWeights(Eigen::MatrixXd &weights, const Eigen::MatrixXd &weightsGrad)=0
This function updates the weights passed based on the selected Optimizer and the weights gradients.