blob: 0f25b21494252f549db2b1b818c684d5a9782fb5 (
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.neuralNetwork import NeuralNetwork
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 = NeuralNetwork(modelFile, boardsize)
#nn = ConvNeuralNetwork(modelFile, boardsize)
nn.trainModel(matches)
nn.saveModel()
if __name__ == '__main__':
main()
|