SYMBOL INDEX (17 symbols across 3 files) FILE: data.py class IndexingTrainDataset (line 8) | class IndexingTrainDataset(Dataset): method __init__ (line 9) | def __init__( method __len__ (line 28) | def __len__(self): method __getitem__ (line 31) | def __getitem__(self, item): class IndexingCollator (line 42) | class IndexingCollator(DataCollatorWithPadding): method __call__ (line 43) | def __call__(self, features): class QueryEvalCollator (line 59) | class QueryEvalCollator(DataCollatorWithPadding): method __call__ (line 60) | def __call__(self, features): FILE: train.py class QueryEvalCallback (line 11) | class QueryEvalCallback(TrainerCallback): method __init__ (line 12) | def __init__(self, test_dataset, logger, restrict_decode_vocab, args: ... method on_epoch_end (line 30) | def on_epoch_end(self, args, state, control, **kwargs): function compute_metrics (line 55) | def compute_metrics(eval_preds): function main (line 69) | def main(): FILE: trainer.py class IndexingTrainer (line 7) | class IndexingTrainer(Trainer): method __init__ (line 8) | def __init__(self, restrict_decode_vocab, **kwds): method compute_loss (line 12) | def compute_loss(self, model, inputs, return_outputs=False): method prediction_step (line 18) | def prediction_step(