causality-aware-retriever

Getting Started

1. Clone the repository

git clone https://github.com/00HS/causality-aware-retrieval.git
cd causality-aware-retrieval

2. Train the model

To train a model, select the appropriate command in run_train.sh by uncommenting it based on your desired model and task.

bash run_train.sh

3. Evaluate the model

bash run_eval.sh

You can compute Hits@k and MRR scores from the saved retrieval results by running score.py after evaluation.