├── README.md ├── citations.md ├── eval ├── run_baselines.py ├── run_eval.py └── run_fava.py ├── fava.png ├── requirements.txt ├── retrieval ├── LICENSE ├── README.md ├── evaluate_retrieved_passages.py ├── example_scripts │ ├── contriever.sh │ └── mcontriever.sh ├── generate_passage_embeddings.py ├── passage_retrieval.py ├── preprocess.py ├── requirements.txt └── src │ ├── __init__.py │ ├── beir_utils.py │ ├── contriever.py │ ├── data.py │ ├── dist_utils.py │ ├── evaluation.py │ ├── finetuning_data.py │ ├── inbatch.py │ ├── index.py │ ├── moco.py │ ├── normalize_text.py │ ├── options.py │ ├── slurm.py │ └── utils.py ├── training ├── generate_train_data.py ├── process_train_data.py └── sample_wiki │ ├── README.md │ ├── build_db.py │ ├── doc_db.py │ └── prepare_source_data.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/README.md -------------------------------------------------------------------------------- /citations.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/citations.md -------------------------------------------------------------------------------- /eval/run_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/eval/run_baselines.py -------------------------------------------------------------------------------- /eval/run_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/eval/run_eval.py -------------------------------------------------------------------------------- /eval/run_fava.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/eval/run_fava.py -------------------------------------------------------------------------------- /fava.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/fava.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/requirements.txt -------------------------------------------------------------------------------- /retrieval/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/LICENSE -------------------------------------------------------------------------------- /retrieval/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/README.md -------------------------------------------------------------------------------- /retrieval/evaluate_retrieved_passages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/evaluate_retrieved_passages.py -------------------------------------------------------------------------------- /retrieval/example_scripts/contriever.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/example_scripts/contriever.sh -------------------------------------------------------------------------------- /retrieval/example_scripts/mcontriever.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/example_scripts/mcontriever.sh -------------------------------------------------------------------------------- /retrieval/generate_passage_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/generate_passage_embeddings.py -------------------------------------------------------------------------------- /retrieval/passage_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/passage_retrieval.py -------------------------------------------------------------------------------- /retrieval/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/preprocess.py -------------------------------------------------------------------------------- /retrieval/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/requirements.txt -------------------------------------------------------------------------------- /retrieval/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /retrieval/src/beir_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/beir_utils.py -------------------------------------------------------------------------------- /retrieval/src/contriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/contriever.py -------------------------------------------------------------------------------- /retrieval/src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/data.py -------------------------------------------------------------------------------- /retrieval/src/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/dist_utils.py -------------------------------------------------------------------------------- /retrieval/src/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/evaluation.py -------------------------------------------------------------------------------- /retrieval/src/finetuning_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/finetuning_data.py -------------------------------------------------------------------------------- /retrieval/src/inbatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/inbatch.py -------------------------------------------------------------------------------- /retrieval/src/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/index.py -------------------------------------------------------------------------------- /retrieval/src/moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/moco.py -------------------------------------------------------------------------------- /retrieval/src/normalize_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/normalize_text.py -------------------------------------------------------------------------------- /retrieval/src/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/options.py -------------------------------------------------------------------------------- /retrieval/src/slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/slurm.py -------------------------------------------------------------------------------- /retrieval/src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/retrieval/src/utils.py -------------------------------------------------------------------------------- /training/generate_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/training/generate_train_data.py -------------------------------------------------------------------------------- /training/process_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/training/process_train_data.py -------------------------------------------------------------------------------- /training/sample_wiki/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/training/sample_wiki/README.md -------------------------------------------------------------------------------- /training/sample_wiki/build_db.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/training/sample_wiki/build_db.py -------------------------------------------------------------------------------- /training/sample_wiki/doc_db.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/training/sample_wiki/doc_db.py -------------------------------------------------------------------------------- /training/sample_wiki/prepare_source_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/training/sample_wiki/prepare_source_data.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abhika-m/FAVA/HEAD/utils.py --------------------------------------------------------------------------------