show_match_nn.py

from ia.predictors import FilePredictor
from ia.random_mcts_player import NNPlayer, fight, RandomMCTSPlayer
from janggi.parameters import N_ITERATIONS, DEFAULT_N_SIMULATIONS
from janggi.utils import Color

# Example of command:
#    python3 show_match_nn.py --number_simulations 800 --n_iterations 200 --root_file_inference /tmp/showmatch --parallel_mcts True --n_threads_mcts 10

player_blue = NNPlayer(Color.BLUE, n_simulations=DEFAULT_N_SIMULATIONS,
                       janggi_net=FilePredictor(),
                       temperature_start=0.01,
                       temperature_threshold=30,
                       temperature_end=0.01,
                       print_info=True)

player_red = RandomMCTSPlayer(Color.RED, n_simulations=16000, temperature_start=0.01,
                              temperature_threshold=30, temperature_end=0.01,
                              print_info=True)

# player_red = NNPlayer(Color.RED, n_simulations=DEFAULT_N_SIMULATIONS,
#                       janggi_net=FilePredictor(),
#                       temperature_start=0.01,
#                       temperature_threshold=30,
#                       temperature_end=0.01,
#                       print_info=True)

fight(player_blue, player_red, N_ITERATIONS, print_board=True)