Esempio n. 1
0
    print(
        f"DQN:     [{mancala1}]  {dqn}\nPlayer:       {player}  [{mancala2}]")


PATH_DQN = "Replay_dqn_state_dict.pt"
DQN_model = DQN_replay(14, 6)
DQN_model.model.load_state_dict(torch.load(PATH_DQN))

game = Game()
done = False

while not done:

    # Player 1 Move
    board1 = game.board()
    game._player_one = True
    p1_action = DQN_model._move(game)
    print(f"DQN Action: {p1_action + 1}")
    game.move(p1_action)
    # End game if move meets win condition
    if game.over():
        break

    # Player 2 Move
    render(game.board())
    game._player_one = False
    p2_action = input("Enter an Action between 1 and 6: ")
    p2_action = 13 - (int(p2_action))
    print(p2_action)
    game.move(p2_action)
    render(game.board())