diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/brain/neural_network.cpp | 4 | ||||
| -rw-r--r-- | src/move_string_mapper.cpp | 2 | ||||
| -rw-r--r-- | src/spacebot.cpp | 2 |
3 files changed, 5 insertions, 3 deletions
diff --git a/src/brain/neural_network.cpp b/src/brain/neural_network.cpp index bacdbed..0c23771 100644 --- a/src/brain/neural_network.cpp +++ b/src/brain/neural_network.cpp @@ -6,7 +6,7 @@ NeuralNetwork::NeuralNetwork(std::istream &&networkConfigFile, unsigned int numb { _neurons.reserve(400); _sensors.reserve(numberOfSensors); - _sensors.reserve(numberOfOutputs); + _outputs.reserve(numberOfOutputs); _biasNode = std::make_shared<BiasNode>(); @@ -28,7 +28,7 @@ NeuralNetwork::NeuralNetwork(std::istream &&networkConfigFile, std::vector<bool> { _neurons.reserve(400); _sensors.reserve(sensorInitialValues.size()); - _sensors.reserve(numberOfOutputs); + _outputs.reserve(numberOfOutputs); _biasNode = std::make_shared<BiasNode>(); diff --git a/src/move_string_mapper.cpp b/src/move_string_mapper.cpp index feffbaa..ed510c1 100644 --- a/src/move_string_mapper.cpp +++ b/src/move_string_mapper.cpp @@ -11,6 +11,8 @@ MoveStringMapper::MoveStringMapper() moveMap[Move::MOVE_RIGHT] = "MoveRight"; moveMap[Move::SHOOT] = "Shoot"; moveMap[Move::BUILD_MISSILE_CONTROLLER] = "BuildMissileController"; + moveMap[Move::BUILD_ALIEN_FACTORY] = "BuildAlienFactory"; + moveMap[Move::BUILD_SHIELD] = "BuildShield"; } std::string MoveStringMapper::toString(const Move &move) diff --git a/src/spacebot.cpp b/src/spacebot.cpp index e18078b..15f2221 100644 --- a/src/spacebot.cpp +++ b/src/spacebot.cpp @@ -23,7 +23,7 @@ Move Spacebot::chooseMove() NeuralNetwork network(std::ifstream(_brainFilename), sensorInputs, - 5); + 7); return static_cast<Move>(network.findMaxOutputIndex()); } |