diff options
author | Justin Worthe <justin.worthe@gmail.com> | 2015-07-27 20:56:04 +0200 |
---|---|---|
committer | Justin Worthe <justin.worthe@gmail.com> | 2015-07-27 20:56:04 +0200 |
commit | 650a2f680ec403d8b2a044674e296b9cab7af793 (patch) | |
tree | caa88427a7c12aca0d3770d1647bee441334a0b9 /include/brain/neural_network.h | |
parent | fc6933f34f7ce4efb522e4af34f94e9e437ec31c (diff) |
Added test and abstract neural node
Diffstat (limited to 'include/brain/neural_network.h')
-rw-r--r-- | include/brain/neural_network.h | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/include/brain/neural_network.h b/include/brain/neural_network.h index d62112c..99e5f60 100644 --- a/include/brain/neural_network.h +++ b/include/brain/neural_network.h @@ -4,17 +4,23 @@ #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 findOutputIndex(); + int findMaxOutputIndex(); + + int numberOfSensors(); + int numberOfOutputs(); private: - std::vector<std::shared_ptr<Neuron>> _neurons; - std::vector<std::shared_ptr<SensorNeuron>> _sensors; - std::vector<std::shared_ptr<Neuron>> _outputs; + std::vector<std::shared_ptr<NeuralNode>> _nodes; + std::vector<std::shared_ptr<Sensor>> _sensors; + std::vector<std::shared_ptr<NeuralNode>> _outputs; }; |