Improved efficiency of network

1 files changed, 10 insertions, 10 deletions

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