-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathsearch
executable file
·20 lines (15 loc) · 1.99 KB
/
search
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#!/usr/bin/env python3
import argparse
import subprocess
print("Searching...")
parser = argparse.ArgumentParser()
parser.add_argument("--collection", required=True, type=str, help="the name of the collection")
parser.add_argument("--topic", required=True, type=str, help="the name of the topic file")
parser.add_argument("--topic_format", default="TREC", type=str, help="the format of the topic file")
args = parser.parse_args()
subprocess.run("java -cp anserini.jar io.anserini.search.SearchCollection -topicreader Trec -index {0} -topics /input/topics/{1} -output /output/run.robust04.bm25.topics.robust04.301-450.601-700.txt -bm25".format(args.collection, args.topic).split())
subprocess.run("java -cp anserini.jar io.anserini.search.SearchCollection -topicreader Trec -index {0} -topics /input/topics/{1} -output /output/run.robust04.bm25+rm3.topics.robust04.301-450.601-700.txt -bm25 -rm3".format(args.collection, args.topic).split())
subprocess.run("java -cp anserini.jar io.anserini.search.SearchCollection -topicreader Trec -index {0} -topics /input/topics/{1} -output /output/run.robust04.bm25+ax.topics.robust04.301-450.601-700.txt -bm25 -axiom -rerankCutoff 20 -axiom.deterministic".format(args.collection, args.topic).split())
subprocess.run("java -cp anserini.jar io.anserini.search.SearchCollection -topicreader Trec -index {0} -topics /input/topics/{1} -output /output/run.robust04.ql.topics.robust04.301-450.601-700.txt -ql".format(args.collection, args.topic).split())
subprocess.run("java -cp anserini.jar io.anserini.search.SearchCollection -topicreader Trec -index {0} -topics /input/topics/{1} -output /output/run.robust04.ql+rm3.topics.robust04.301-450.601-700.txt -ql -rm3".format(args.collection, args.topic).split())
subprocess.run("java -cp anserini.jar io.anserini.search.SearchCollection -topicreader Trec -index {0} -topics /input/topics/{1} -output /output/run.robust04.ql+ax.topics.robust04.301-450.601-700.txt -ql -axiom -rerankCutoff 20 -axiom.deterministic".format(args.collection, args.topic).split())