Skip to content

Commit 3dbbc56

Browse files
Update param.py
1 parent 2be48da commit 3dbbc56

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/param.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,9 @@
99
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
1010

1111
settings = {
12-
'cmd': ['rag', 'search'], # steps of pipeline, ['query_refinement', 'similarity', 'rag', 'search', 'rag_fusion', 'eval', 'agg', 'build', 'box','dense_retrieve', 'stats]
12+
'cmd': ['query_refinement', 'similarity', 'search', 'eval', 'agg'], # steps of pipeline, ['query_refinement', 'similarity', 'search', 'eval', 'agg', 'build', 'box','dense_retrieve', 'stats]
1313
'datalist': ['./../data/raw/robust04', './../data/raw/gov2', './../data/raw/antique', './../data/raw/dbpedia', './../data/raw/clueweb09b'], # ['./../data/raw/robust04', './../data/raw/gov2', './../data/raw/antique', './../data/raw/dbpedia', './../data/raw/clueweb09b']
1414
'domainlist': ['robust04', 'gov2', 'antique', 'dbpedia', 'clueweb09b'], # ['robust04', 'gov2', 'antique', 'dbpedia', 'clueweb09b']
15-
'fusion_category': ['global', 'local'], # ['all', 'global', 'local', 'bt_nllb', 'bt']
16-
'fusion_method': 'rrf_multi_k', # 'rrf_multi_k, rrf', 'condorcet', 'random'
1715
'ncore': 2,
1816
'ranker': ['bm25'], # ['qld', 'bm25', 'tct_colbert']
1917
'metric': ['map'], # any valid trec_eval.9.0.4 metrics like ['map', 'ndcg', 'recip_rank']

0 commit comments

Comments
 (0)