diff options
| -rwxr-xr-x | testKeras.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/testKeras.py b/testKeras.py index 0f25b21..0f518d0 100755 --- a/testKeras.py +++ b/testKeras.py @@ -5,7 +5,7 @@ import sys from imago.sgfParser.sgf import loadGameTree -from imago.engine.keras.neuralNetwork import NeuralNetwork +from imago.engine.keras.denseNeuralNetwork import DenseNeuralNetwork from imago.engine.keras.convNeuralNetwork import ConvNeuralNetwork def main(): @@ -18,7 +18,7 @@ def main(): modelFile = "" boardsize = 9 - nn = NeuralNetwork(modelFile, boardsize) + nn = DenseNeuralNetwork(modelFile, boardsize) #nn = ConvNeuralNetwork(modelFile, boardsize) nn.trainModel(matches) nn.saveModel() |