53 lines
1.6 KiB
C++
53 lines
1.6 KiB
C++
#include "layer.h"
|
|
#include "matrices.h"
|
|
#include <vector>
|
|
|
|
class Network {
|
|
public:
|
|
Matrix input;
|
|
float (*activation)(float) = Layer::Sigmoid; // Activation function is sigmoid by default
|
|
|
|
std::vector<Layer> hidden_layers;
|
|
Layer output_layer;
|
|
|
|
inline void Feed(Matrix);
|
|
inline Matrix GetOutput();
|
|
|
|
inline void Forward();
|
|
|
|
// Constructors
|
|
// Input size, Array of hidden sizes, Output size
|
|
Network(int, std::vector<int>, int);
|
|
};
|
|
|
|
Network::Network(int input_size, std::vector<int> hidden_sizes, int output_size){
|
|
this->input = Matrix(input_size, 1);
|
|
|
|
this->hidden_layers.push_back(Layer(input_size, hidden_sizes[0]));
|
|
for(int i = 1; i < hidden_sizes.size(); i++){
|
|
// For every hidden layer, create a layer of specified size
|
|
this->hidden_layers.push_back(Layer(hidden_sizes[i-1], hidden_sizes[i]));
|
|
}
|
|
this->output_layer = Layer(hidden_sizes[hidden_sizes.size() - 1], output_size);
|
|
}
|
|
|
|
Matrix Network::GetOutput(){
|
|
return this->output_layer.activated_output;
|
|
}
|
|
|
|
void Network::Feed(Matrix a){
|
|
this->input = a;
|
|
}
|
|
|
|
void Network::Forward(){
|
|
// Feeding first layer
|
|
this->hidden_layers[0].Feed(this->input);
|
|
this->hidden_layers[0].Forward();
|
|
for(int i = 1; i < this->hidden_layers.size(); i++){
|
|
// Feeding A(L-1) and forwarding
|
|
this->hidden_layers[i].Feed(this->hidden_layers[i - 1].activated_output);
|
|
this->hidden_layers[i].Forward();
|
|
}
|
|
this->output_layer.Feed(this->hidden_layers[this->hidden_layers.size() - 1].activated_output);
|
|
this->output_layer.Forward();
|
|
} |