def name(self):
return "TreeSearch"
def version(self):
return "1.0"
def pick_move(self, color):
x,y = choose_move_alphabeta(self.board, self.policy, self.value, depth=3)
return Move(x,y)
def get_position_eval(self):
return self.value.evaluate(self.board)
if __name__ == '__main__':
import GTP
fclient = GTP.redirect_all_output("log_engine.txt")
import Policy
import MoveModels
import Eval
import EvalModels
#policy = Policy.AllPolicy()
policy = Policy.TFPolicy(model=MoveModels.Conv12PosDepELU(N=19, Nfeat=21), threshold_prob=0.8, softmax_temp=1.0)
value = Eval.TFEval(EvalModels.Conv11PosDepFC1ELU(N=19, Nfeat=21))
engine = TreeSearchEngine(policy, value)
gtp = GTP.GTP(engine, fclient)
gtp.loop()
