SYMBOL INDEX (265 symbols across 20 files)

FILE: qa-eval/distributed_run.py
  function distributed_main (line 6) | def distributed_main(args):
  function worker (line 19) | def worker(gpu, ngpus_per_node, args):

FILE: qa-eval/main.py
  class HarvestingQADataset (line 15) | class HarvestingQADataset(Dataset):
    method __init__ (line 16) | def __init__(self, filename, ratio):
    method __getitem__ (line 20) | def __getitem__(self, idx):
    method __len__ (line 38) | def __len__(self):
  function main (line 42) | def main(args):

FILE: qa-eval/squad_utils.py
  class SquadExample (line 16) | class SquadExample(object):
    method __init__ (line 21) | def __init__(self,
    method __str__ (line 37) | def __str__(self):
    method __repr__ (line 40) | def __repr__(self):
  class InputFeatures (line 55) | class InputFeatures(object):
    method __init__ (line 57) | def __init__(self,
  function convert_examples_to_features (line 102) | def convert_examples_to_features(examples, tokenizer, max_seq_length,
  function convert_examples_to_harv_features (line 268) | def convert_examples_to_harv_features(examples, tokenizer, max_seq_length,
  function convert_examples_to_features_answer_id (line 399) | def convert_examples_to_features_answer_id(examples, tokenizer, max_seq_...
  function read_examples (line 610) | def read_examples(input_file, debug=False, is_training=False):
  function read_squad_examples (line 666) | def read_squad_examples(input_file, is_training, version_2_with_negative...
  function _improve_answer_span (line 746) | def _improve_answer_span(doc_tokens, input_start, input_end, tokenizer,
  function _check_is_max_context (line 783) | def _check_is_max_context(doc_spans, cur_span_index, position):
  function write_predictions (line 821) | def write_predictions(all_examples, all_features, all_results, n_best_size,
  function get_final_text (line 1027) | def get_final_text(pred_text, orig_text, do_lower_case, verbose_logging=...
  function _get_best_indexes (line 1122) | def _get_best_indexes(logits, n_best_size):
  function _compute_softmax (line 1135) | def _compute_softmax(scores):
  function write_answer_predictions (line 1158) | def write_answer_predictions(all_examples, all_features, all_results, n_...
  function normalize_answer (line 1359) | def normalize_answer(s):
  function f1_score (line 1378) | def f1_score(prediction, ground_truth):
  function exact_match_score (line 1392) | def exact_match_score(prediction, ground_truth):
  function metric_max_over_ground_truths (line 1396) | def metric_max_over_ground_truths(metric_fn, prediction, ground_truths):
  function evaluate (line 1404) | def evaluate(dataset, predictions):
  function read_predictions (line 1428) | def read_predictions(prediction_file):
  function read_answers (line 1434) | def read_answers(gold_file):
  function evaluate_mrqa (line 1446) | def evaluate_mrqa(answers, predictions, skip_no_answer=False):

FILE: qa-eval/trainer.py
  function get_linear_schedule_with_warmup (line 30) | def get_linear_schedule_with_warmup(optimizer, num_warmup_steps, num_tra...
  class Trainer (line 41) | class Trainer(object):
    method __init__ (line 42) | def __init__(self, args):
    method make_model_env (line 47) | def make_model_env(self, gpu, ngpus_per_node):
    method get_pretrain_loader (line 94) | def get_pretrain_loader(self):
    method get_dev_loader (line 102) | def get_dev_loader(self):
    method get_test_loader (line 109) | def get_test_loader(self):
    method train (line 116) | def train(self):
    method evaluate_model (line 180) | def evaluate_model(self, msg, dev=True):
    method set_random_seed (line 252) | def set_random_seed(self, random_seed=2019):

FILE: qa-eval/utils.py
  function time_since (line 5) | def time_since(t):
  function progress_bar (line 10) | def progress_bar(completed, total, step=5):
  function user_friendly_time (line 32) | def user_friendly_time(s):
  function eta (line 53) | def eta(start, completed, total):
  function cal_running_avg_loss (line 64) | def cal_running_avg_loss(loss, running_avg_loss, decay=0.99):
  function kl_coef (line 72) | def kl_coef(i):
  function compute_kernel (line 79) | def compute_kernel(x, y):
  function compute_mmd (line 91) | def compute_mmd(x, y):
  class EMA (line 99) | class EMA(object):
    method __init__ (line 106) | def __init__(self, model, decay):
    method __call__ (line 116) | def __call__(self, model, num_updates):
    method assign (line 125) | def assign(self, model):
    method resume (line 137) | def resume(self, model):

FILE: vae/eval.py
  function to_string (line 13) | def to_string(index, tokenizer):
  class Result (line 29) | class Result(object):
    method __init__ (line 30) | def __init__(self,
  function eval_vae (line 51) | def eval_vae(epoch, args, trainer, eval_data):

FILE: vae/generate_qa.py
  class CustomDatset (line 13) | class CustomDatset(Dataset):
    method __init__ (line 14) | def __init__(self, tokenizer, input_file, max_length=512):
    method __getitem__ (line 20) | def __getitem__(self, idx):
    method __len__ (line 32) | def __len__(self):
  function main (line 36) | def main(args):

FILE: vae/main.py
  function main (line 15) | def main(args):

FILE: vae/models.py
  function return_mask_lengths (line 10) | def return_mask_lengths(ids):
  function cal_attn (line 16) | def cal_attn(query, memories, mask):
  function gumbel_softmax (line 25) | def gumbel_softmax(logits, tau=1, hard=False, eps=1e-20, dim=-1):
  class CategoricalKLLoss (line 44) | class CategoricalKLLoss(nn.Module):
    method __init__ (line 45) | def __init__(self):
    method forward (line 48) | def forward(self, P, Q):
  class GaussianKLLoss (line 55) | class GaussianKLLoss(nn.Module):
    method __init__ (line 56) | def __init__(self):
    method forward (line 59) | def forward(self, mu1, logvar1, mu2, logvar2):
  class Embedding (line 66) | class Embedding(nn.Module):
    method __init__ (line 67) | def __init__(self, bert_model):
    method forward (line 76) | def forward(self, input_ids, token_type_ids=None, position_ids=None):
  class ContextualizedEmbedding (line 96) | class ContextualizedEmbedding(nn.Module):
    method __init__ (line 97) | def __init__(self, bert_model):
    method forward (line 104) | def forward(self, input_ids, attention_mask, token_type_ids=None):
  class CustomLSTM (line 128) | class CustomLSTM(nn.Module):
    method __init__ (line 129) | def __init__(self, input_size, hidden_size, num_layers, dropout, bidir...
    method forward (line 142) | def forward(self, inputs, input_lengths, state=None):
  class PosteriorEncoder (line 158) | class PosteriorEncoder(nn.Module):
    method __init__ (line 159) | def __init__(self, embedding, emsize,
    method forward (line 187) | def forward(self, c_ids, q_ids, a_ids):
  class PriorEncoder (line 244) | class PriorEncoder(nn.Module):
    method __init__ (line 245) | def __init__(self, embedding, emsize,
    method forward (line 269) | def forward(self, c_ids):
    method interpolation (line 294) | def interpolation(self, c_ids, zq):
  class AnswerDecoder (line 316) | class AnswerDecoder(nn.Module):
    method __init__ (line 317) | def __init__(self, embedding, emsize,
    method forward (line 334) | def forward(self, init_state, c_ids):
    method generate (line 353) | def generate(self, init_state, c_ids):
  class ContextEncoderforQG (line 383) | class ContextEncoderforQG(nn.Module):
    method __init__ (line 384) | def __init__(self, embedding, emsize,
    method forward (line 398) | def forward(self, c_ids, a_ids):
  class QuestionDecoder (line 414) | class QuestionDecoder(nn.Module):
    method __init__ (line 415) | def __init__(self, sos_id, eos_id,
    method postprocess (line 456) | def postprocess(self, q_ids):
    method forward (line 471) | def forward(self, init_state, c_ids, q_ids, a_ids):
    method generate (line 535) | def generate(self, init_state, c_ids, a_ids):
    method sample (line 589) | def sample(self, init_state, c_ids, a_ids):
  class DiscreteVAE (line 644) | class DiscreteVAE(nn.Module):
    method __init__ (line 645) | def __init__(self, args):
    method return_init_state (line 712) | def return_init_state(self, zq, za):
    method forward (line 728) | def forward(self, c_ids, q_ids, a_ids, start_positions, end_positions):
    method generate (line 779) | def generate(self, zq, za, c_ids):
    method return_answer_logits (line 789) | def return_answer_logits(self, zq, za, c_ids):

FILE: vae/qgevalcap/bleu/bleu.py
  function precook (line 33) | def precook(s, n=4, out=False):
  function cook_refs (line 45) | def cook_refs(refs, eff=None, n=4): ## lhuang: oracle will call with "av...
  function cook_test (line 70) | def cook_test(test, reflen, refmaxcounts, eff=None, n=4):
  class BleuScorer (line 95) | class BleuScorer(object):
    method copy (line 102) | def copy(self):
    method __init__ (line 110) | def __init__(self, test=None, refs=None, n=4, special_reflen=None):
    method cook_append (line 119) | def cook_append(self, test, refs):
    method ratio (line 132) | def ratio(self, option=None):
    method score_ratio (line 136) | def score_ratio(self, option=None):
    method score_ratio_str (line 140) | def score_ratio_str(self, option=None):
    method reflen (line 143) | def reflen(self, option=None):
    method testlen (line 147) | def testlen(self, option=None):
    method retest (line 151) | def retest(self, new_test):
    method rescore (line 162) | def rescore(self, new_test):
    method size (line 167) | def size(self):
    method __iadd__ (line 171) | def __iadd__(self, other):
    method compatible (line 185) | def compatible(self, other):
    method single_reflen (line 188) | def single_reflen(self, option="average"):
    method _single_reflen (line 191) | def _single_reflen(self, reflens, option=None, testlen=None):
    method recompute_score (line 204) | def recompute_score(self, option=None, verbose=0):
    method compute_score (line 208) | def compute_score(self, option=None, verbose=0):
  class Bleu (line 277) | class Bleu:
    method __init__ (line 278) | def __init__(self, n=4):
    method compute_score (line 284) | def compute_score(self, gts, res):
    method method (line 309) | def method(self):

FILE: vae/qgevalcap/bleu/bleu_scorer.py
  function precook (line 23) | def precook(s, n=4, out=False):
  function cook_refs (line 35) | def cook_refs(refs, eff=None, n=4): ## lhuang: oracle will call with "av...
  function cook_test (line 60) | def cook_test(test, (reflen, refmaxcounts), eff=None, n=4):
  class BleuScorer (line 85) | class BleuScorer(object):
    method copy (line 92) | def copy(self):
    method __init__ (line 100) | def __init__(self, test=None, refs=None, n=4, special_reflen=None):
    method cook_append (line 109) | def cook_append(self, test, refs):
    method ratio (line 122) | def ratio(self, option=None):
    method score_ratio (line 126) | def score_ratio(self, option=None):
    method score_ratio_str (line 130) | def score_ratio_str(self, option=None):
    method reflen (line 133) | def reflen(self, option=None):
    method testlen (line 137) | def testlen(self, option=None):
    method retest (line 141) | def retest(self, new_test):
    method rescore (line 152) | def rescore(self, new_test):
    method size (line 157) | def size(self):
    method __iadd__ (line 161) | def __iadd__(self, other):
    method compatible (line 175) | def compatible(self, other):
    method single_reflen (line 178) | def single_reflen(self, option="average"):
    method _single_reflen (line 181) | def _single_reflen(self, reflens, option=None, testlen=None):
    method recompute_score (line 194) | def recompute_score(self, option=None, verbose=0):
    method compute_score (line 198) | def compute_score(self, option=None, verbose=0):

FILE: vae/qgevalcap/cider/cider.py
  function precook (line 21) | def precook(s, n=4, out=False):
  function cook_refs (line 38) | def cook_refs(refs, n=4): ## lhuang: oracle will call with "average"
  function cook_test (line 48) | def cook_test(test, n=4):
  class CiderScorer (line 57) | class CiderScorer(object):
    method copy (line 61) | def copy(self):
    method __init__ (line 68) | def __init__(self, test=None, refs=None, n=4, sigma=6.0):
    method cook_append (line 78) | def cook_append(self, test, refs):
    method size (line 88) | def size(self):
    method __iadd__ (line 92) | def __iadd__(self, other):
    method compute_doc_freq (line 103) | def compute_doc_freq(self):
    method compute_cider (line 116) | def compute_cider(self):
    method compute_score (line 193) | def compute_score(self, option=None, verbose=0):
  class Cider (line 203) | class Cider:
    method __init__ (line 208) | def __init__(self, test=None, refs=None, n=4, sigma=6.0):
    method compute_score (line 214) | def compute_score(self, gts, res):
    method method (line 243) | def method(self):

FILE: vae/qgevalcap/cider/cider_scorer.py
  function precook (line 11) | def precook(s, n=4, out=False):
  function cook_refs (line 28) | def cook_refs(refs, n=4): ## lhuang: oracle will call with "average"
  function cook_test (line 38) | def cook_test(test, n=4):
  class CiderScorer (line 47) | class CiderScorer(object):
    method copy (line 51) | def copy(self):
    method __init__ (line 58) | def __init__(self, test=None, refs=None, n=4, sigma=6.0):
    method cook_append (line 68) | def cook_append(self, test, refs):
    method size (line 78) | def size(self):
    method __iadd__ (line 82) | def __iadd__(self, other):
    method compute_doc_freq (line 93) | def compute_doc_freq(self):
    method compute_cider (line 106) | def compute_cider(self):
    method compute_score (line 183) | def compute_score(self, option=None, verbose=0):

FILE: vae/qgevalcap/eval.py
  function precook (line 11) | def precook(s, n=4, out=False):
  function cook_refs (line 24) | def cook_refs(refs, eff=None, n=4):  ## lhuang: oracle will call with "a...
  function cook_test (line 50) | def cook_test(test, reflen, refmaxcounts, eff=None, n=4):
  class BleuScorer (line 76) | class BleuScorer(object):
    method copy (line 84) | def copy(self):
    method __init__ (line 92) | def __init__(self, test=None, refs=None, n=4, special_reflen=None):
    method cook_append (line 101) | def cook_append(self, test, refs):
    method ratio (line 114) | def ratio(self, option=None):
    method score_ratio (line 118) | def score_ratio(self, option=None):
    method score_ratio_str (line 122) | def score_ratio_str(self, option=None):
    method reflen (line 125) | def reflen(self, option=None):
    method testlen (line 129) | def testlen(self, option=None):
    method retest (line 133) | def retest(self, new_test):
    method rescore (line 144) | def rescore(self, new_test):
    method size (line 149) | def size(self):
    method __iadd__ (line 153) | def __iadd__(self, other):
    method compatible (line 167) | def compatible(self, other):
    method single_reflen (line 170) | def single_reflen(self, option="average"):
    method _single_reflen (line 173) | def _single_reflen(self, reflens, option=None, testlen=None):
    method recompute_score (line 186) | def recompute_score(self, option=None, verbose=0):
    method compute_score (line 190) | def compute_score(self, option=None, verbose=0):
  class Bleu (line 258) | class Bleu:
    method __init__ (line 259) | def __init__(self, n=4):
    method compute_score (line 265) | def compute_score(self, gts, res):
    method method (line 289) | def method(self):
  class QGEvalCap (line 293) | class QGEvalCap:
    method __init__ (line 294) | def __init__(self, gts, res):
    method evaluate (line 298) | def evaluate(self, not_print=True):
  function eval_qg (line 325) | def eval_qg(res_dict, gts_dict, not_print=True):

FILE: vae/qgevalcap/meteor/meteor.py
  class Meteor (line 15) | class Meteor:
    method __init__ (line 17) | def __init__(self):
    method compute_score (line 33) | def compute_score(self, gts, res):
    method method (line 53) | def method(self):
    method _stat (line 56) | def _stat(self, hypothesis_str, reference_list):
    method _score (line 64) | def _score(self, hypothesis_str, reference_list):
    method __del__ (line 81) | def __del__(self):

FILE: vae/qgevalcap/rouge/rouge.py
  function my_lcs (line 13) | def my_lcs(string, sub):
  class Rouge (line 36) | class Rouge():
    method __init__ (line 41) | def __init__(self):
    method calc_score (line 45) | def calc_score(self, candidate, refs):
    method compute_score (line 77) | def compute_score(self, gts, res):
    method method (line 104) | def method(self):

FILE: vae/squad_utils.py
  class SquadExample (line 17) | class SquadExample(object):
    method __init__ (line 22) | def __init__(self,
    method __str__ (line 38) | def __str__(self):
    method __repr__ (line 41) | def __repr__(self):
  class InputFeatures (line 56) | class InputFeatures(object):
    method __init__ (line 58) | def __init__(self,
  function convert_examples_to_features (line 103) | def convert_examples_to_features(examples, tokenizer, max_seq_length,
  function convert_examples_to_harv_features (line 269) | def convert_examples_to_harv_features(examples, tokenizer, max_seq_length,
  function convert_examples_to_features_answer_id (line 400) | def convert_examples_to_features_answer_id(examples, tokenizer, max_seq_...
  function read_examples (line 611) | def read_examples(input_file, debug=False, is_training=False):
  function read_squad_examples (line 667) | def read_squad_examples(input_file, is_training, version_2_with_negative...
  function _improve_answer_span (line 747) | def _improve_answer_span(doc_tokens, input_start, input_end, tokenizer,
  function _check_is_max_context (line 784) | def _check_is_max_context(doc_spans, cur_span_index, position):
  function write_predictions (line 822) | def write_predictions(all_examples, all_features, all_results, n_best_size,
  function get_final_text (line 1028) | def get_final_text(pred_text, orig_text, do_lower_case, verbose_logging=...
  function _get_best_indexes (line 1123) | def _get_best_indexes(logits, n_best_size):
  function _compute_softmax (line 1136) | def _compute_softmax(scores):
  function write_answer_predictions (line 1159) | def write_answer_predictions(all_examples, all_features, all_results, n_...
  function normalize_answer (line 1360) | def normalize_answer(s):
  function f1_score (line 1379) | def f1_score(prediction, ground_truth):
  function exact_match_score (line 1393) | def exact_match_score(prediction, ground_truth):
  function metric_max_over_ground_truths (line 1397) | def metric_max_over_ground_truths(metric_fn, prediction, ground_truths):
  function evaluate (line 1405) | def evaluate(dataset, predictions):
  function read_predictions (line 1429) | def read_predictions(prediction_file):
  function read_answers (line 1435) | def read_answers(gold_file):
  function evaluate_mrqa (line 1447) | def evaluate_mrqa(answers, predictions, skip_no_answer=False):

FILE: vae/trainer.py
  class VAETrainer (line 7) | class VAETrainer(object):
    method __init__ (line 8) | def __init__(self, args):
    method train (line 23) | def train(self, c_ids, q_ids, a_ids, start_positions, end_positions):
    method generate_posterior (line 46) | def generate_posterior(self, c_ids, q_ids, a_ids):
    method generate_answer_logits (line 53) | def generate_answer_logits(self, c_ids, q_ids, a_ids):
    method generate_prior (line 60) | def generate_prior(self, c_ids):
    method save (line 67) | def save(self, filename):

FILE: vae/translate.py
  function return_mask_lengths (line 14) | def return_mask_lengths(ids):
  function post_process (line 20) | def post_process(q_ids, start_positions, end_positions, c_ids, total_max...
  function main (line 63) | def main(args):

FILE: vae/utils.py
  function get_squad_data_loader (line 11) | def get_squad_data_loader(tokenizer, file, shuffle, args):
  function get_harv_data_loader (line 33) | def get_harv_data_loader(tokenizer, file, shuffle, ratio, args):
  function batch_to_device (line 50) | def batch_to_device(batch, device):