aboutsummaryrefslogtreecommitdiff
path: root/train.py
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()