From f35c5fca6a9faaa1caf3c1992844adf9033744e8 Mon Sep 17 00:00:00 2001
From: Justin Worthe <justin.worthe@gmail.com>
Date: Sat, 29 Aug 2015 20:22:40 +0200
Subject: Improved efficiency of network

---
 include/brain/neural_network.h | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

(limited to 'include/brain/neural_network.h')

diff --git a/include/brain/neural_network.h b/include/brain/neural_network.h
index f75fdea..a1af7cc 100644
--- a/include/brain/neural_network.h
+++ b/include/brain/neural_network.h
@@ -14,25 +14,25 @@
 class NeuralNetwork
 {
 public:
-    NeuralNetwork(std::istream &&networkConfigFile, int numberOfSensors, int numberOfOutputs);
-    NeuralNetwork(std::istream &&networkConfigFile, std::vector<bool> sensorInitialValues, int numberOfOutputs);
+    NeuralNetwork(std::istream &&networkConfigFile, unsigned int numberOfSensors, unsigned int numberOfOutputs);
+    NeuralNetwork(std::istream &&networkConfigFile, std::vector<bool> sensorInitialValues, unsigned int numberOfOutputs);
     
-    void setInput(int inputIndex, double activation);
-    int findMaxOutputIndex() const;
+    void setInput(unsigned int inputIndex, double activation);
+    unsigned int findMaxOutputIndex() const;
 
-    int numberOfSensors() const { return _sensors.size(); }
-    int numberOfOutputs() const { return _outputs.size(); }
+    unsigned int numberOfSensors() const { return _sensors.size(); }
+    unsigned int numberOfOutputs() const { return _outputs.size(); }
 	
 private:
-    std::map<int, std::shared_ptr<Sensor>> _sensors;
+    std::vector<std::shared_ptr<Sensor>> _sensors;
     std::shared_ptr<BiasNode> _biasNode;
-    std::map<int, std::shared_ptr<Neuron>> _neurons;
+    std::vector<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);
+    std::shared_ptr<Sensor> findOrAddSensor(unsigned int id);
+    std::shared_ptr<Neuron> findOrAddNeuron(unsigned int id);
 };
 
-- 
cgit v1.2.3