From 259a6e8a1ff030060ca624d07f37f3ffd0c2295d Mon Sep 17 00:00:00 2001 From: InigoGutierrez Date: Wed, 22 Jun 2022 21:33:19 +0200 Subject: Fixed using wrong network for testing. --- testKeras.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'testKeras.py') 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() -- cgit v1.2.1