SYMBOL INDEX (26 symbols across 3 files)

FILE: inference.py
  function load_test_dataset (line 13) | def load_test_dataset():

FILE: train.py
  function _make_r_io_base (line 36) | def _make_r_io_base(f, mode: str):
  function jload (line 41) | def jload(f, mode="r"):
  class ModelArguments (line 68) | class ModelArguments:
  class DataArguments (line 73) | class DataArguments:
  class TrainingArguments (line 79) | class TrainingArguments(transformers.TrainingArguments):
  function safe_save_model_for_hf_trainer (line 92) | def safe_save_model_for_hf_trainer(trainer: transformers.Trainer, output...
  function smart_tokenizer_and_embedding_resize (line 101) | def smart_tokenizer_and_embedding_resize(
  function _tokenize_fn (line 124) | def _tokenize_fn(strings: Sequence[str], tokenizer: transformers.PreTrai...
  class ScoreDataset (line 148) | class ScoreDataset(Dataset):
    method __init__ (line 151) | def __init__(self, data_path: str, tokenizer: transformers.PreTrainedT...
    method __len__ (line 158) | def __len__(self):
    method __getitem__ (line 161) | def __getitem__(self, i):
  function _single_tokenize (line 164) | def _single_tokenize(text, tokenizer, max_len=None):
  function stop_response (line 176) | def stop_response(res):
  class DataCollatorForSupervisedDataset (line 184) | class DataCollatorForSupervisedDataset(object):
    method __call__ (line 190) | def __call__(self, instances):
  function make_supervised_data_module (line 233) | def make_supervised_data_module(tokenizer: transformers.PreTrainedTokeni...
  class AlignmentTrainer (line 240) | class AlignmentTrainer(Trainer):
    method gather_logits_labels (line 241) | def gather_logits_labels(self, logits, labels):
    method get_score (line 250) | def get_score(self, logit_label, labels):
    method align_loss (line 256) | def align_loss(self, scores, idxs, rw_scores):
    method sft_loss (line 270) | def sft_loss(self, logit_label, idxs, rw_scores):
    method compute_loss (line 274) | def compute_loss(self, model, inputs, return_outputs=False):
  function train (line 300) | def train():

FILE: utils/retrieve_relative_triple.py
  function clean_kg (line 7) | def clean_kg(triples):