diff options
Diffstat (limited to 'include/brain/neural_network.h')
| -rw-r--r-- | include/brain/neural_network.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/include/brain/neural_network.h b/include/brain/neural_network.h index 22e5e42..f75fdea 100644 --- a/include/brain/neural_network.h +++ b/include/brain/neural_network.h @@ -26,8 +26,8 @@ public: private: std::map<int, std::shared_ptr<Sensor>> _sensors; std::shared_ptr<BiasNode> _biasNode; - std::map<int, std::shared_ptr<Neuron>> _hiddenNodes; - std::map<int, std::shared_ptr<Neuron>> _outputs; + std::map<int, std::shared_ptr<Neuron>> _neurons; + std::vector<std::shared_ptr<Neuron>> _outputs; void parseFile(std::istream &&file); |