Types & classes — github.com/AkariAsai/self-rag

CodeHub MCPcopy Create free account

hub / github.com/AkariAsai/self-rag / types & classes

Types & classes23 in github.com/AkariAsai/self-rag

⨍ Functions 256◇ Types & classes 23

↓ 1 callersClassDataCollatorForSupervisedDataset

Collate examples for supervised fine-tuning.

data_creation/train_special_tokens.py:258

↓ 1 callersClassDataset

Monolingual dataset based on a list of paths

retrieval_lm/src/data.py:84

↓ 1 callersClassDenseEncoderModel

retrieval_lm/src/beir_utils.py:22

↓ 1 callersClassMultiDataset

retrieval_lm/src/data.py:53

↓ 1 callersClassRetriever

retrieval_lm/passage_retrieval.py:30

↓ 1 callersClassSimpleTokenizer

retrieval_lm/src/evaluation.py:22

↓ 1 callersClassSupervisedDataset

Dataset for supervised fine-tuning.

data_creation/train_special_tokens.py:227

retrieval_lm/src/finetuning_data.py:127

retrieval_lm/src/data.py:115

ClassContriever

retrieval_lm/src/contriever.py:11

ClassCosineScheduler

retrieval_lm/src/utils.py:108

ClassDataArguments

data_creation/train_special_tokens.py:74

retrieval_lm/src/finetuning_data.py:11

retrieval_lm/src/dist_utils.py:7

retrieval_lm/src/inbatch.py:17

retrieval_lm/src/index.py:15

retrieval_lm/src/moco.py:14

ClassModelArguments

data_creation/train_special_tokens.py:45

retrieval_lm/src/options.py:7

ClassTrainingArguments

data_creation/train_special_tokens.py:84

ClassWarmupLinearScheduler

retrieval_lm/src/utils.py:91

ClassWeightedAvgStats

provides an average over a bunch of stats

retrieval_lm/src/utils.py:153

ClassXLMRetriever

retrieval_lm/src/contriever.py:58