From 650a2f680ec403d8b2a044674e296b9cab7af793 Mon Sep 17 00:00:00 2001
From: Justin Worthe <justin.worthe@gmail.com>
Date: Mon, 27 Jul 2015 20:56:04 +0200
Subject: Added test and abstract neural node

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

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

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;
 };
 
-- 
cgit v1.2.3