#!/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()