Source code for gameanalysis.script.gplearn

"""Module for constructing game models by Gaussian process regression."""
import argparse
import json
import pickle
import sys

from gameanalysis import gameio
from gameanalysis import gpgame


[docs]def add_parser(subparsers): parser = subparsers.add_parser( 'gplearn', help="""Learn a gp model""", description="""Learn a gp model""") parser.add_argument( '--input', '-i', metavar='<input-file>', default=sys.stdin, type=argparse.FileType('r'), help="""Input file for script. (default: stdin)""") parser.add_argument( '--output', '-o', metavar='<output-file>', default=sys.stdout.buffer, type=argparse.FileType('wb'), help="""Output file for script. (default: stdout)""") return parser
[docs]def main(args): data, serial = gameio.read_game(json.load(args.input)) game = gpgame.BaseGPGame(data) pickle.dump(game, args.output) pickle.dump(serial, args.output)