Changed network innards to use maps

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);