blob: 308cb56c6ffeef12007f5e1387bfb143cc1838d9 (
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
|
#pragma once
#include <memory>
#include <istream>
#include <vector>
#include "brain/neural_node.h"
#include "brain/sensor.h"
class NeuralNetwork
{
public:
NeuralNetwork(std::istream &&networkConfigFile, int numberOfSensors, 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::vector<std::shared_ptr<NeuralNode>> _nodes;
std::vector<std::shared_ptr<Sensor>> _sensors;
std::vector<std::shared_ptr<NeuralNode>> _outputs;
};
|