diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/brain/neural_network.h | 3 | ||||
-rw-r--r-- | include/spacebot.h | 4 |
2 files changed, 4 insertions, 3 deletions
diff --git a/include/brain/neural_network.h b/include/brain/neural_network.h index b2c441f..ccd65eb 100644 --- a/include/brain/neural_network.h +++ b/include/brain/neural_network.h @@ -14,7 +14,8 @@ class NeuralNetwork { public: NeuralNetwork(std::istream &&networkConfigFile, int numberOfSensors, int numberOfOutputs); - + NeuralNetwork(std::istream &&networkConfigFile, std::vector<bool> sensorInitialValues, int numberOfOutputs); + void setInput(int inputIndex, double activation); int findMaxOutputIndex() const; diff --git a/include/spacebot.h b/include/spacebot.h index 079b33e..9f7f83e 100644 --- a/include/spacebot.h +++ b/include/spacebot.h @@ -6,11 +6,11 @@ class Spacebot { public: - Spacebot(std::string outputPath); + Spacebot(std::string outputPath, std::string brainFilename); void writeNextMove(); private: std::string _outputFilename; - std::string _networkConfigFilename; + std::string _brainFilename; GameState _gameState; void writeMove(const Move& move); |