blob: 0f518d070830209634431f04ff0d6bd7528c3358 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
#!/usr/bin/python
"""Starts training a keras neural network."""
import sys
from imago.sgfParser.sgf import loadGameTree
from imago.engine.keras.denseNeuralNetwork import DenseNeuralNetwork
from imago.engine.keras.convNeuralNetwork import ConvNeuralNetwork
def main():
games = []
for file in sys.argv[1:]:
print(file)
games.append(loadGameTree(file))
matches = [game.getMainLineOfPlay() for game in games]
modelFile = ""
boardsize = 9
nn = DenseNeuralNetwork(modelFile, boardsize)
#nn = ConvNeuralNetwork(modelFile, boardsize)
nn.trainModel(matches)
nn.saveModel()
if __name__ == '__main__':
main()
|
