diff options
Diffstat (limited to 'testKeras.py')
-rwxr-xr-x | testKeras.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/testKeras.py b/testKeras.py index 3b0b3f5..0f25b21 100755 --- a/testKeras.py +++ b/testKeras.py @@ -6,6 +6,7 @@ import sys from imago.sgfParser.sgf import loadGameTree from imago.engine.keras.neuralNetwork import NeuralNetwork +from imago.engine.keras.convNeuralNetwork import ConvNeuralNetwork def main(): games = [] @@ -15,8 +16,10 @@ def main(): matches = [game.getMainLineOfPlay() for game in games] - modelFile = "models/testModel.h5" - nn = NeuralNetwork(modelFile, 9) + modelFile = "" + boardsize = 9 + nn = NeuralNetwork(modelFile, boardsize) + #nn = ConvNeuralNetwork(modelFile, boardsize) nn.trainModel(matches) nn.saveModel() |