diff options
Diffstat (limited to 'include/brain/neural_network.h')
-rw-r--r-- | include/brain/neural_network.h | 14 |
1 files changed, 10 insertions, 4 deletions
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; }; |