gomoku_rl.utils package

Submodules

• gomoku_rl.utils.augment module
  • AntiDiagonalFlip
    • AntiDiagonalFlip.inverse_map_board()
    • AntiDiagonalFlip.inverse_map_index()
    • AntiDiagonalFlip.map_board()
    • AntiDiagonalFlip.map_index()
  • DiagonalFlip
    • DiagonalFlip.inverse_map_board()
    • DiagonalFlip.inverse_map_index()
    • DiagonalFlip.map_board()
    • DiagonalFlip.map_index()
  • HorizontalFlip
    • HorizontalFlip.inverse_map_board()
    • HorizontalFlip.inverse_map_index()
    • HorizontalFlip.map_board()
    • HorizontalFlip.map_index()
  • Identity
    • Identity.inverse_map_board()
    • Identity.inverse_map_index()
    • Identity.map_board()
    • Identity.map_index()
  • Rotation
    • Rotation.inverse_map_board()
    • Rotation.inverse_map_index()
    • Rotation.map_board()
    • Rotation.map_index()
  • Transform
    • Transform.inverse_map_board()
    • Transform.inverse_map_index()
    • Transform.map_board()
    • Transform.map_index()
  • VerticalFlip
    • VerticalFlip.inverse_map_board()
    • VerticalFlip.inverse_map_index()
    • VerticalFlip.map_board()
    • VerticalFlip.map_index()
  • augment_transition()
  • get_augmented_transition()
• gomoku_rl.utils.elo module
  • Elo
    • Elo.addPlayer()
    • Elo.expected_score()
    • Elo.update()
  • compute_elo_ratings()
  • compute_expected_score()
• gomoku_rl.utils.eval module
  • eval_win_rate()
  • get_payoff_matrix()
• gomoku_rl.utils.log module
  • Mean
    • Mean.update()
    • Mean.value
  • get_log_func()
• gomoku_rl.utils.misc module
  • add_prefix()
  • get_kwargs()
  • set_seed()
• gomoku_rl.utils.module module
  • ActorNet
    • ActorNet.forward()
  • MyDuelingCnnDQNet
    • MyDuelingCnnDQNet.forward()
  • PolicyHead
    • PolicyHead.forward()
  • ResidualBlock
    • ResidualBlock.forward()
  • ResidualTower
    • ResidualTower.forward()
  • ValueHead
    • ValueHead.forward()
  • ValueNet
    • ValueNet.forward()
  • count_parameters()
  • init_params()
• gomoku_rl.utils.policy module
  • uniform_policy()
• gomoku_rl.utils.psro module
  • ConvergedIndicator
    • ConvergedIndicator.converged()
    • ConvergedIndicator.reset()
    • ConvergedIndicator.update()
  • PSROPolicyWrapper
    • PSROPolicyWrapper.add_current_policy()
    • PSROPolicyWrapper.eval()
    • PSROPolicyWrapper.sample()
    • PSROPolicyWrapper.set_meta_policy()
    • PSROPolicyWrapper.set_oracle_mode()
    • PSROPolicyWrapper.train()
  • PayoffType
    • PayoffType.black_vs_white
    • PayoffType.both
  • Population
    • Population.add()
    • Population.fixed_behavioural_strategy()
    • Population.make_behavioural_strategy()
    • Population.sample()
  • calculate_jpc()
  • get_meta_solver()
  • get_new_payoffs()
  • get_new_payoffs_sp()
  • init_payoffs_sp()
  • print_payoffs()
  • solve_last_n()
  • solve_nash()
  • solve_uniform()
• gomoku_rl.utils.sampler module
  • SequentialSampler
    • SequentialSampler.ran_out
    • SequentialSampler.sample()
• gomoku_rl.utils.test module
  • Type
    • Type.black
    • Type.mixed
    • Type.white
  • assert_layer_transition()
  • assert_observation()
  • assert_tensor_1d_all()
  • assert_transition()
  • no_nan_in_tensordict()
• gomoku_rl.utils.visual module
  • annotate_heatmap()
  • heatmap()
  • payoff_headmap()
• gomoku_rl.utils.wandb module
  • dict_flatten()
  • init_wandb()

Module contents