def _create_neural_mcts_player(
    game: polygames.Game,
    mcts_option: mcts.MctsOption,
    num_actor: int,
    actor_channel: tube.DataChannel,
    model_manager: Optional[polygames.ModelManager] = None,
    rnn_state_shape: List[int] = [],
    rnn_seqlen: int = 0,
    logit_value: bool = False,
) -> mcts.MctsPlayer:

    player = mcts.MctsPlayer(mcts_option)
    for _ in range(num_actor):
        num_actor += 1
        actor = polygames.Actor(
            actor_channel,
            game.get_feat_size(),
            game.get_action_size(),
            rnn_state_shape,
            rnn_seqlen,
            logit_value,
            True,
            True,
            model_manager,
        )
        player.set_actor(actor)
    return player
Exemplo n.º 2
0
def _create_pure_mcts_player(game: polygames.Game,
                             mcts_option: mcts.MctsOption,
                             num_actor: int) -> mcts.MctsPlayer:
    """a player that uses only mcts + random rollout, no neural net"""
    player = mcts.MctsPlayer(mcts_option)
    for _ in range(num_actor):
        actor = polygames.Actor(None, game.get_feat_size(),
                                game.get_action_size(), False, False, None)
        player.add_actor(actor)
    return player
def _create_neural_mcts_player(
    game: polygames.Game,
    mcts_option: mcts.MctsOption,
    num_actor: int,
    actor_channel: tube.DataChannel,
    assembler: Optional[tube.ChannelAssembler] = None,
) -> mcts.MctsPlayer:
    player = mcts.MctsPlayer(mcts_option)
    for _ in range(num_actor):
        num_actor += 1
        actor = polygames.Actor(
            actor_channel,
            game.get_feat_size(),
            game.get_action_size(),
            True,
            True,
            assembler,
        )
        player.add_actor(actor)
    return player
def _create_forward_player(
    game: polygames.Game,
    actor_channel: tube.DataChannel,
    model_manager: Optional[polygames.ModelManager] = None,
    rnn_state_shape: List[int] = [],
    rnn_seqlen: int = 0,
    logit_value: bool = False,
) -> mcts.MctsPlayer:

    player = polygames.ForwardPlayer()
    actor = polygames.Actor(
        actor_channel,
        game.get_feat_size(),
        game.get_action_size(),
        rnn_state_shape,
        rnn_seqlen,
        logit_value,
        True,
        True,
        model_manager,
    )
    player.set_actor(actor)
    return player