diff options
Diffstat (limited to 'include/brain')
-rw-r--r-- | include/brain/neural_network.h | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/include/brain/neural_network.h b/include/brain/neural_network.h index ccd65eb..22e5e42 100644 --- a/include/brain/neural_network.h +++ b/include/brain/neural_network.h @@ -3,6 +3,7 @@ #include <memory> #include <istream> #include <vector> +#include <map> #include <string> #include "brain/neural_node.h" @@ -23,10 +24,10 @@ public: int numberOfOutputs() const { return _outputs.size(); } private: - std::vector<std::shared_ptr<Sensor>> _sensors; + std::map<int, std::shared_ptr<Sensor>> _sensors; std::shared_ptr<BiasNode> _biasNode; - std::vector<std::shared_ptr<Neuron>> _hiddenNodes; - std::vector<std::shared_ptr<Neuron>> _outputs; + std::map<int, std::shared_ptr<Neuron>> _hiddenNodes; + std::map<int, std::shared_ptr<Neuron>> _outputs; void parseFile(std::istream &&file); |