# !/bin/bash | |
datasets=("mag" "amazon" "prime") | |
# Define scorer_name mapping using an associative array | |
declare -A dataset_scorer_map=( | |
[mag]="ada" | |
[amazon]="ada" | |
[prime]="contriever" | |
) | |
for dataset in "${datasets[@]}"; do | |
# Get the corresponding scorer_name for the dataset | |
scorer_name="${dataset_scorer_map[$dataset]}" | |
echo "Processing dataset: $dataset with scorer: $scorer_name" | |
python eval.py --dataset_name "$dataset" --scorer_name "$scorer_name" --mod "test" | |
cd Reranking | |
python rerank.py --dataset_name "$dataset" | |
cd .. | |
done | |