blob: f75fdea69625353a22d8fa56e104e6f5e43fab99 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#pragma once
#include <memory>
#include <istream>
#include <vector>
#include <map>
#include <string>
#include "brain/neural_node.h"
#include "brain/sensor.h"
#include "brain/bias_node.h"
#include "brain/neuron.h"
class NeuralNetwork
{
public:
NeuralNetwork(std::istream &&networkConfigFile, int numberOfSensors, int numberOfOutputs);
NeuralNetwork(std::istream &&networkConfigFile, std::vector<bool> sensorInitialValues, int numberOfOutputs);
void setInput(int inputIndex, double activation);
int findMaxOutputIndex() const;
int numberOfSensors() const { return _sensors.size(); }
int numberOfOutputs() const { return _outputs.size(); }
private:
std::map<int, std::shared_ptr<Sensor>> _sensors;
std::shared_ptr<BiasNode> _biasNode;
std::map<int, std::shared_ptr<Neuron>> _neurons;
std::vector<std::shared_ptr<Neuron>> _outputs;
void parseFile(std::istream &&file);
void addLink(std::shared_ptr<NeuralNode> source, std::shared_ptr<Neuron> destination, double weight);
std::shared_ptr<Sensor> findOrAddSensor(int id);
std::shared_ptr<Neuron> findOrAddNeuron(int id);
};
|