SYMBOL INDEX (104 symbols across 20 files) FILE: src/data/mlm.py class MLMDataset (line 9) | class MLMDataset(Dataset): method __init__ (line 10) | def __init__(self, data): method __len__ (line 15) | def __len__(self): method __getitem__ (line 18) | def __getitem__(self, idx): FILE: src/data/policy_value.py class PolicyValueDataset (line 5) | class PolicyValueDataset(Dataset): method __init__ (line 6) | def __init__(self, data): method __len__ (line 9) | def __len__(self): method __getitem__ (line 12) | def __getitem__(self, idx): FILE: src/features/common.py function get_seq_from_board (line 6) | def get_seq_from_board(board): FILE: src/features/policy_value.py function get_move_label (line 12) | def get_move_label(move, color): function get_result (line 64) | def get_result(result, color): function get_policy_value_label (line 75) | def get_policy_value_label(hcpe): function get_policy_value_label_from_moves (line 87) | def get_policy_value_label_from_moves(moves): function get_moves_from_lines (line 99) | def get_moves_from_lines(line): FILE: src/model/bert.py class BertMLM (line 22) | class BertMLM(nn.Module): method __init__ (line 23) | def __init__(self, model_dir=None): method forward (line 30) | def forward(self, input_ids, labels): class BertPolicyValue (line 34) | class BertPolicyValue(nn.Module): method __init__ (line 35) | def __init__(self, model_dir=None): method forward (line 58) | def forward(self, input_ids, labels=None): FILE: src/pl_modules/__init__.py function get_pl_modules (line 5) | def get_pl_modules(cfg): FILE: src/pl_modules/mlm.py class MLMModule (line 12) | class MLMModule(pl.LightningModule): method __init__ (line 14) | def __init__(self, hparams): method forward (line 19) | def forward(self, batch): method training_step (line 24) | def training_step(self, batch, batch_idx): method validation_step (line 30) | def validation_step(self, batch, batch_idx): method validation_epoch_end (line 35) | def validation_epoch_end(self, outputs): method configure_optimizers (line 40) | def configure_optimizers(self): class MLMDataModule (line 44) | class MLMDataModule(pl.LightningDataModule): method __init__ (line 45) | def __init__(self, cfg): method setup (line 49) | def setup(self, stage=None): method train_dataloader (line 56) | def train_dataloader(self): method val_dataloader (line 59) | def val_dataloader(self): FILE: src/pl_modules/policy_value.py class PolicyValueModule (line 12) | class PolicyValueModule(pl.LightningModule): method __init__ (line 13) | def __init__(self, hparams): method forward (line 18) | def forward(self, input_ids, labels=None): method training_step (line 22) | def training_step(self, batch, batch_idx): method validation_step (line 27) | def validation_step(self, batch, batch_idx): method validation_epoch_end (line 34) | def validation_epoch_end(self, outputs): method configure_optimizers (line 42) | def configure_optimizers(self): class PolicyValueDataModule (line 46) | class PolicyValueDataModule(pl.LightningDataModule): method __init__ (line 47) | def __init__(self, cfg): method setup (line 51) | def setup(self, stage=None): method train_dataloader (line 60) | def train_dataloader(self): method val_dataloader (line 63) | def val_dataloader(self): FILE: src/player/base_player.py class BasePlayer (line 4) | class BasePlayer: method __init__ (line 5) | def __init__(self): method usi (line 8) | def usi(self): method usinewgame (line 12) | def usinewgame(self): method setoption (line 15) | def setoption(self, option): method isready (line 18) | def isready(self): method position (line 21) | def position(self, moves): method go (line 31) | def go(self): method quit (line 34) | def quit(self): FILE: src/player/mcts_player.py class MCTSPlayer (line 18) | class MCTSPlayer(BasePlayer): method __init__ (line 19) | def __init__(self, ckpt_path, playout_halt=1000, temperature=1, resign... method usi (line 33) | def usi(self): method isready (line 37) | def isready(self): method go (line 45) | def go(self): method expand_node (line 123) | def expand_node(self): method eval_node (line 152) | def eval_node(self, n_idx): method uct_search (line 182) | def uct_search(self, n_idx): method select_max_ucb_child (line 221) | def select_max_ucb_child(self, c_idx): method interruption_check (line 235) | def interruption_check(self): function argparse (line 249) | def argparse(): function main (line 258) | def main(args): FILE: src/player/policy_player.py class PolicyPlayer (line 17) | class PolicyPlayer(BasePlayer): method __init__ (line 18) | def __init__(self, ckpt_path): method usi (line 23) | def usi(self): method isready (line 27) | def isready(self): method go (line 34) | def go(self): function argparse (line 72) | def argparse(): function main (line 81) | def main(args): FILE: src/player/usi.py function usi (line 4) | def usi(player: BasePlayer): FILE: src/uct/uct_node.py function hash_to_index (line 8) | def hash_to_index(zhash): class NodeHashEntry (line 12) | class NodeHashEntry: method __init__ (line 13) | def __init__(self): method reset (line 19) | def reset(self): class NodeHash (line 26) | class NodeHash: method __init__ (line 27) | def __init__(self): method initialize (line 32) | def initialize(self): method search_empty_index (line 43) | def search_empty_index(self, zhash, color, moves): method find_same_hash_index (line 65) | def find_same_hash_index(self, zhash, color, moves): method save_used_hash (line 78) | def save_used_hash(self, board, uct_nodes, n_idx): method delete_old_hash (line 93) | def delete_old_hash(self, board, uct_node): method get_usage_rate (line 106) | def get_usage_rate(self): class UctNode (line 110) | class UctNode: method __init__ (line 111) | def __init__(self): method reset (line 121) | def reset(self): FILE: src/utils/hcpe.py function get_data_from_hcpe (line 8) | def get_data_from_hcpe(hcpes): function load_hcpes (line 17) | def load_hcpes(hcpe_paths): FILE: src/utils/misc.py function greedy (line 4) | def greedy(logits): function boltzmann (line 8) | def boltzmann(logits, temperature): FILE: src/utils/sfen.py function get_gokaku_sfen_paths (line 8) | def get_gokaku_sfen_paths(base_dir: Path, max_num_of_moves=40): function get_data_from_sfen (line 19) | def get_data_from_sfen(sfens): function load_sfen (line 36) | def load_sfen(sfen_path): function load_sfens (line 42) | def load_sfens(sfen_paths): FILE: tools/make_dataset.py function main (line 10) | def main(): FILE: tools/pl_to_transformers.py function argparse (line 9) | def argparse(): function main (line 16) | def main(args): FILE: tools/test_engine.py function parse_args (line 7) | def parse_args(): function main (line 13) | def main(args): FILE: tools/train.py function argparse (line 12) | def argparse(): function main (line 22) | def main(args):