diff options
Diffstat (limited to 'imago/engine/keras/keras.py')
-rw-r--r-- | imago/engine/keras/keras.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/imago/engine/keras/keras.py b/imago/engine/keras/keras.py index a6aa913..80b6647 100644 --- a/imago/engine/keras/keras.py +++ b/imago/engine/keras/keras.py @@ -3,16 +3,18 @@ from imago.gameLogic.gameMove import GameMove from imago.gameLogic.gameBoard import GameBoard from imago.engine.decisionAlgorithm import DecisionAlgorithm -from imago.engine.keras.neuralNetwork import NeuralNetwork +from imago.engine.keras.denseNeuralNetwork import DenseNeuralNetwork +from imago.engine.keras.convNeuralNetwork import ConvNeuralNetwork -MODEL_FILE = "models/testModel.h5" +MODEL_FILE = "" # Use network's default model file class Keras(DecisionAlgorithm): def __init__(self, move): self.currentMove = move self.boardSize = move.board.getBoardHeight() - self.nn = NeuralNetwork(MODEL_FILE, self.boardSize) + #self.nn = NeuralNetwork(MODEL_FILE, self.boardSize) + self.nn = ConvNeuralNetwork(MODEL_FILE, self.boardSize) def forceNextMove(self, coords): """Selects given move as next move.""" |