-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathreproduce_msmarco_evaluation.sh
executable file
·39 lines (34 loc) · 1.6 KB
/
reproduce_msmarco_evaluation.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#!/usr/bin/env bash
# BFS
echo "BFS 1-Hop" >> temp.tsv
src/evaluate_bfs.py --dataset msmarco --path_len 1 >> temp.tsv
echo "BFS 2-Hop" >> temp.tsv
src/evaluate_bfs.py --dataset msmarco --path_len 2 >> temp.tsv
echo "BFS 3-Hop" >> temp.tsv
src/evaluate_bfs.py --dataset msmarco --path_len 3 >> temp.tsv
echo "BFS 4-Hop" >> temp.tsv
src/evaluate_bfs.py --dataset msmarco --path_len 4 >> temp.tsv
# UnifiedQA
echo "UnifiedQA-v2" >> temp.tsv
src/evaluate_unifiedqa.py --dataset msmarco \
--context_file "src/predictions/msmarco3_context.jsonl" \
>> temp.tsv
echo "UnifiedQA-v2-T" >> temp.tsv
src/evaluate_unifiedqa.py --dataset msmarco --use_context \
--context_file "src/predictions/msmarco_simple_context.jsonl" \
>> temp.tsv
echo "UnifiedQA-v2-P" >> temp.tsv
src/evaluate_unifiedqa.py --dataset msmarco --use_context \
--context_file "src/predictions/msmarco3_context.jsonl" \
>> temp.tsv
# Agent
echo "Agent 1-Hop" >> temp.tsv
src/evaluate_agent.py --dataset msmarco --model "models/msmarco_1.pt" --path_len_eval 1 >> temp.tsv
echo "Agent 2-Hop" >> temp.tsv
src/evaluate_agent.py --dataset msmarco --model "models/msmarco_2.pt" --path_len_eval 2 >> temp.tsv
echo "Agent 3-Hop" >> temp.tsv
src/evaluate_agent.py --dataset msmarco --model "models/msmarco_3.pt" --path_len_eval 3 >> temp.tsv
echo "Agent 4-Hop" >> temp.tsv
src/evaluate_agent.py --dataset msmarco --model "models/msmarco_4.pt" --path_len_eval 4 >> temp.tsv
./build_md.py temp.tsv "msmarco_evaluation.md"
rm temp.tsv