# !/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