File size: 570 Bytes
f96a150 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# !/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
|