diff options
| author | InigoGutierrez <inigogf.95@gmail.com> | 2022-06-22 21:33:19 +0200 |
|---|---|---|
| committer | InigoGutierrez <inigogf.95@gmail.com> | 2022-06-22 21:33:19 +0200 |
| commit | 259a6e8a1ff030060ca624d07f37f3ffd0c2295d (patch) | |
| tree | f7382b43a07cc711d36da7bda6c1c05bf9d7deaf | |
| parent | 206f2cf2592f1ea36d28e1026c5bfb0a6432fbd2 (diff) | |
| download | imago-259a6e8a1ff030060ca624d07f37f3ffd0c2295d.tar.gz imago-259a6e8a1ff030060ca624d07f37f3ffd0c2295d.zip | |
Fixed using wrong network for testing.
| -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() |