SYMBOL INDEX (118 symbols across 19 files)

FILE: Baselines/BuildOrderPrediction/test.py
  function calc_action_acc (line 10) | def calc_action_acc(action_pre, action_gt):
  function calc_weighted_action_acc (line 13) | def calc_weighted_action_acc(action_pre, action_gt, weight):
  function show_test_result (line 16) | def show_test_result(name, phrase, result, steps=10, title=''):

FILE: Baselines/BuildOrderPrediction/train.py
  class BuildOrderGRU (line 23) | class BuildOrderGRU(torch.nn.Module):
    method __init__ (line 24) | def __init__(self, num_inputs, num_outputs):
    method forward (line 36) | def forward(self, states, require_init):
    method detach (line 62) | def detach(self):
  function train (line 68) | def train(model, env, args):
  function test (line 184) | def test(model, env, args):
  function next_path (line 238) | def next_path(model_folder, paths):
  function main (line 249) | def main():

FILE: Baselines/BuildOrderPrediction/train_spatial.py
  class BuildOrderGRU (line 23) | class BuildOrderGRU(torch.nn.Module):
    method __init__ (line 24) | def __init__(self, n_channels, n_features, n_actions):
    method forward (line 39) | def forward(self, states_S, states_G, require_init):
    method detach (line 69) | def detach(self):
  function train (line 73) | def train(model, env, args):
  function test (line 191) | def test(model, env, args):
  function next_path (line 248) | def next_path(model_folder, paths):
  function main (line 259) | def main():

FILE: Baselines/GlobalStateEvaluation/test.py
  function calc_value_acc (line 10) | def calc_value_acc(value_pre, value_gt):
  function calc_weighted_value_acc (line 13) | def calc_weighted_value_acc(value_pre, value_gt, weight):
  function show_test_result (line 16) | def show_test_result(name, phrase, result, steps=10, title=''):

FILE: Baselines/GlobalStateEvaluation/train.py
  class StateEvaluationGRU (line 23) | class StateEvaluationGRU(torch.nn.Module):
    method __init__ (line 24) | def __init__(self, num_inputs):
    method forward (line 36) | def forward(self, states, require_init):
    method detach (line 62) | def detach(self):
  function train (line 68) | def train(model, env, args):
  function test (line 180) | def test(model, env, args):
  function next_path (line 235) | def next_path(model_folder, paths):
  function main (line 246) | def main():

FILE: Baselines/GlobalStateEvaluation/train_spatial.py
  class StateEvaluationGRU (line 23) | class StateEvaluationGRU(torch.nn.Module):
    method __init__ (line 24) | def __init__(self, n_channels, n_features):
    method forward (line 39) | def forward(self, states_S, states_G, require_init):
    method detach (line 69) | def detach(self):
  function train (line 73) | def train(model, env, args):
  function test (line 188) | def test(model, env, args):
  function next_path (line 246) | def next_path(model_folder, paths):
  function main (line 257) | def main():

FILE: data_loader/BatchEnv.py
  class BatchEnv (line 10) | class BatchEnv(object):
    method __init__ (line 11) | def __init__(self):
    method init (line 14) | def init(self, path, root, race, enemy_race, step_mul=8, n_replays=4, ...
    method __generate_replay_list__ (line 42) | def __generate_replay_list__(self, replays, race):
    method __init_epoch__ (line 45) | def __init_epoch__(self):
    method __reset__ (line 59) | def __reset__(self):
    method __load_replay__ (line 70) | def __load_replay__(self, path):
    method step (line 73) | def step(self, **kwargs):
    method __one_step__ (line 100) | def __one_step__(self, replay_dict, done):
    method __post_process__ (line 103) | def __post_process__(self, result, **kwargs):
    method step_count (line 106) | def step_count(self):
    method close (line 109) | def close(self):
  class BatchGlobalFeatureEnv (line 115) | class BatchGlobalFeatureEnv(BatchEnv):
    method __post_init__ (line 121) | def __post_init__(self):
    method __generate_replay_list__ (line 125) | def __generate_replay_list__(self, replays, root, race):
    method __load_replay__ (line 133) | def __load_replay__(self, path):
    method __one_step__ (line 141) | def __one_step__(self, replay_dict, done):
    method __post_process__ (line 156) | def __post_process__(self, result, reward=True, action=False, score=Fa...
  class BatchSpatialEnv (line 169) | class BatchSpatialEnv(BatchEnv):
    method __post_init__ (line 175) | def __post_init__(self):
    method __generate_replay_list__ (line 178) | def __generate_replay_list__(self, replays, root, race):
    method __load_replay__ (line 186) | def __load_replay__(self, path):
    method __one_step__ (line 195) | def __one_step__(self, replay_dict, done):
    method __post_process__ (line 213) | def __post_process__(self, result, reward=True, action=False, score=Fa...

FILE: extract_features/SpatialFeatures.py
  class ScreenFeatures (line 6) | class ScreenFeatures(collections.namedtuple("ScreenFeatures", ["height_m...
    method __new__ (line 11) | def __new__(cls, **kwargs):
  class MinimapFeatures (line 26) | class MinimapFeatures(collections.namedtuple("MinimapFeatures", [
    method __new__ (line 31) | def __new__(cls, **kwargs):
  class SpatialFeatures (line 67) | class SpatialFeatures(Features):
    method observation_spec (line 68) | def observation_spec(self):
    method transform_obs (line 87) | def transform_obs(self, obs):

FILE: extract_features/game_state.py
  function load_stat (line 6) | def load_stat(path):
  class GameState (line 26) | class GameState(object):
    method __init__ (line 42) | def __init__(self, stat_path, enemy_stat_path):
    method update (line 66) | def update(self, state):
    method get_action (line 84) | def get_action(self):
    method __set_units__ (line 87) | def __set_units__(self, units):
    method __set_action__ (line 100) | def __set_action__(self, action):
    method __units2vec__ (line 109) | def __units2vec__(self, units, stat):
    method __set_to_array__ (line 145) | def __set_to_array__(self, set_var, key2id):
    method __dict_to_array__ (line 151) | def __dict_to_array__(self, dict_var, key2id, scale=1):
    method to_vector (line 157) | def to_vector(self):
    method __str__ (line 177) | def __str__(self):

FILE: extract_features/global_feature_vector.py
  function parse_replay (line 26) | def parse_replay(replay_player_path, reward, race, enemy_race):
  function main (line 42) | def main():

FILE: extract_features/replay_stat.py
  function update (line 29) | def update(replay_path, stat):
  function post_process (line 65) | def post_process(stat):
  function main (line 80) | def main():

FILE: extract_features/spatial_feature_tensor.py
  function parse_replay (line 34) | def parse_replay(replay_player_path, sampled_action_path, reward, race, ...
  class Parser (line 81) | class Parser(object):
    method __init__ (line 82) | def __init__(self, race_vs_race, races, stats):
    method __call__ (line 87) | def __call__(self, line):
  function main (line 108) | def main():

FILE: extract_features/split.py
  function save (line 23) | def save(replays, prefix, folder):
  function main (line 28) | def main():

FILE: parse_replay/extract_actions.py
  class ReplayProcessor (line 46) | class ReplayProcessor(multiprocessing.Process):
    method __init__ (line 48) | def __init__(self, run_config, replay_queue, counter, total_num):
    method run (line 55) | def run(self):
    method process_replay (line 91) | def process_replay(self, controller, replay_data, map_data, player_id,...
  function replay_queue_filler (line 112) | def replay_queue_filler(replay_queue, replay_list):
  function main (line 117) | def main():

FILE: parse_replay/parse_replay.py
  class ReplayProcessor (line 47) | class ReplayProcessor(multiprocessing.Process):
    method __init__ (line 49) | def __init__(self, run_config, replay_queue, counter, total_num):
    method run (line 56) | def run(self):
    method process_replay (line 116) | def process_replay(self, controller, replay_data, map_data, player_id,...
  function replay_queue_filler (line 134) | def replay_queue_filler(replay_queue, replay_list):
  function main (line 139) | def main():

FILE: parse_replay/replay2global_features.py
  function process_replay (line 27) | def process_replay(sampled_action, actions, observations, feat, units_in...
  function parse_replay (line 116) | def parse_replay(replay_player_path, sampled_action_path, reward):
  function main (line 148) | def main():

FILE: parse_replay/sample_actions.py
  function sample_action_from_player (line 28) | def sample_action_from_player(action_path):
  function sample_action (line 54) | def sample_action(replay_path, action_path, sampled_path):
  function main (line 79) | def main():

FILE: preprocess/parse_replay_info.py
  class ReplayProcessor (line 31) | class ReplayProcessor(multiprocessing.Process):
    method __init__ (line 33) | def __init__(self, run_config, replay_queue, counter, total_num):
    method run (line 40) | def run(self):
  function replay_queue_filler (line 62) | def replay_queue_filler(replay_queue, replay_list):
  function main (line 67) | def main():

FILE: preprocess/preprocess.py
  function valid_replay (line 31) | def valid_replay(info, ping):
  function main (line 54) | def main():