#!/usr/bin/python

"""Starts training a keras neural network."""

import sys

from imago.sgfParser.sgf import loadGameTree
from imago.engine.keras.neuralNetwork import NeuralNetwork

def main():
    games = []
    for file in sys.argv[1:]:
        print(file)
        games.append(loadGameTree(file))

    matches = [game.getMainLineOfPlay() for game in games]

    modelFile = "models/testModel.h5"
    nn = NeuralNetwork(modelFile, 9)
    nn.trainModel(matches)
    nn.saveModel()

if __name__ == '__main__':
    main()