Nogo.py

#!/usr/bin/env python3
from nogo.gtp_connection import GtpConnection, format_point, point_to_coord
from nogo.board_util import GoBoardUtil, BLACK, WHITE, EMPTY, BORDER, coord_to_point
from nogo.simple_board import SimpleGoBoard
from MCTS import MCTS
from nogo.pytorch.NNet import NNetWrapper as nn
from nogo.NogoGame import NogoGame
import numpy as np
from utils import *
import sys

class Nogo():
    def __init__(self):
        """
        NoGo player for alpha-zero-general

        """
        self.name = "NoGoAlphaZeroGeneral"
        self.version = 1.0

        self.g = NogoGame(7)
        self.n1 = nn(self.g)
        self.n1.load_checkpoint('./pretrained_models/nogo7x7/','best.pth.tar')
        args1 = dotdict({'numMCTSSims': 200, 'cpuct':1.0})
        self.mcts = MCTS(self.g, self.n1, args1)
        self.n1p = lambda x: np.argmax(self.mcts.getActionProb(x, temp=0))
        
    def get_move(self, board, color):
        self.g.board.current_player = color
        curPlayer = 1 if color == BLACK else -1
        canonicalBoard = self.g.getCanonicalForm(board, curPlayer, False)
        action = self.n1p(canonicalBoard)
        point = self.g.convert_point(action)
        return point
    
def run():
    """
    start the gtp connection and wait for commands.
    """

    con = GtpConnection(Nogo())
    con.start_connection()
    con.play_cmd()

if __name__=='__main__':
    run()