FlexNN.h

Go to the documentation of this file.

 
#ifndef FlexNN_H
#define FlexNN_H
 
#include <vector>
#include <Eigen/Dense>
 
#include "Layer.h"
 
namespace FlexNN
{
  class NeuralNetwork
  {
  public:
    NeuralNetwork(const std::vector<Layer> &layers) : layers(layers) {}
 
    void train(const Eigen::MatrixXd &input, const Eigen::MatrixXd &target, double learningRate, int epochs);
 
    double accuracy(const Eigen::MatrixXd &X, const Eigen::MatrixXd &Y);
 
    Eigen::MatrixXd predict(const Eigen::MatrixXd &input)
    {
      auto outputs = forward(input);
      return outputs.back(); // Return the final output (activation of the last layer)
    }
 
  private:
    std::vector<Layer> layers;
 
    std::vector<Eigen::MatrixXd> forward(const Eigen::MatrixXd &input);
 
    std::vector<Eigen::MatrixXd> backward(const std::vector<Eigen::MatrixXd> &outputs, const Eigen::MatrixXd &target);
 
    void updateWeights(const std::vector<Eigen::MatrixXd> &gradients, double learningRate);
  };
}
 
#endif // FlexNN_H