25 lines
754 B
Python
25 lines
754 B
Python
import chess
|
|
from chesspp.i_strategy import IStrategy
|
|
from chesspp.mcts.classic_mcts_node_v2 import ClassicMctsNodeV2
|
|
|
|
|
|
class ClassicMctsV2:
|
|
def __init__(self, board: chess.Board, color: chess.Color, strategy: IStrategy):
|
|
self.board = board
|
|
self.color = color
|
|
self.strategy = strategy
|
|
self.root = ClassicMctsNodeV2(board, color, strategy)
|
|
|
|
def build_tree(self, samples: int = 1000):
|
|
"""
|
|
Runs the MCTS with the given number of samples
|
|
:param samples: number of simulations
|
|
:return: best node containing the best move
|
|
"""
|
|
for i in range(samples):
|
|
node = self.root._select_leaf()
|
|
score = node._rollout()
|
|
node._backpropagate(score)
|
|
|
|
|