diff options
Diffstat (limited to 'src/spacebot.cpp')
-rw-r--r-- | src/spacebot.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/spacebot.cpp b/src/spacebot.cpp index 418756c..15f2221 100644 --- a/src/spacebot.cpp +++ b/src/spacebot.cpp @@ -2,6 +2,7 @@ #include "move_string_mapper.h" #include "brain/neural_network.h" #include <fstream> +#include <iostream> Spacebot::Spacebot(std::string outputPath, std::string brainFilename) : _outputFilename(outputPath+"/move.txt"), @@ -22,10 +23,9 @@ Move Spacebot::chooseMove() NeuralNetwork network(std::ifstream(_brainFilename), sensorInputs, - static_cast<int>(Move::BUILD_SHIELD)); + 7); - int moveInt = network.findMaxOutputIndex(); - return static_cast<Move>(moveInt); + return static_cast<Move>(network.findMaxOutputIndex()); } void Spacebot::writeMove(const Move& move) |