main.py

import os
import sys
import yaml
import tensorflow as tf
sys.path.insert(0, './src/')

import Blackbird
# from TicTacToe import BoardState
from DragonChess import BoardState


def main():
    if not os.path.isfile('parameters.yaml'):
        raise IOError('Copy parameters_template.yaml into parameters.yaml')
    with open('parameters.yaml') as param_file:
        parameters = yaml.safe_load(param_file)

    physical_devices = tf.config.list_physical_devices('GPU')
    print(f"Num GPUs: {len(physical_devices)}")

    model = Blackbird.Model(BoardState, parameters['name'], parameters.get(
        'mcts'), parameters.get('network'), parameters.get('tensorflow'))

    Blackbird.GenerateTrainingSamples(model,
                                    1,
                                    parameters.get('mcts').get('temperature').get('exploration'))
    Blackbird.TrainWithExamples(model,
                                batchSize=parameters.get('network').get('training').get('batch_size'),
                                learningRate=parameters.get('network').get('training').get('learning_rate'))

    print('Against a random player:')
    print(Blackbird.TestRandom(model,
                               parameters.get('mcts').get(
                                   'temperature').get('exploitation'),
                               10))

    print('Against the last best player:')
    print(Blackbird.TestPrevious(model,
                                 parameters.get('mcts').get(
                                     'temperature').get('exploitation'),
                                 10))

    print('Against a good player:')
    print(Blackbird.TestGood(model,
                             parameters.get('mcts').get(
                                 'temperature').get('exploitation'),
                             10))

    print('\n')
    del model


if __name__ == '__main__':
    main()