mirror of
https://github.com/Sosokker/CPP-Neural-Network.git
synced 2025-12-18 18:14:04 +01:00
23 lines
491 B
C++
23 lines
491 B
C++
#ifndef LAYERS_HPP
|
|
#define LAYERS_HPP
|
|
|
|
#include "../../include/Eigen/Dense"
|
|
|
|
class Layer {
|
|
public:
|
|
Layer();
|
|
|
|
virtual void forward(const Eigen::VectorXd& input_data);
|
|
virtual void backward(const Eigen::VectorXd& output_gradient, double learning_rate);
|
|
|
|
const Eigen::VectorXd& getOutput() const;
|
|
const Eigen::VectorXd& getInputGradient() const;
|
|
|
|
protected:
|
|
Eigen::VectorXd input;
|
|
Eigen::VectorXd output;
|
|
Eigen::VectorXd input_gradient;
|
|
};
|
|
|
|
#endif // LAYERS_HPP
|