|
9 | 9 | os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
|
10 | 10 |
|
11 | 11 | 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] |
13 | 13 | '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']
|
14 | 14 | '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' |
17 | 15 | 'ncore': 2,
|
18 | 16 | 'ranker': ['bm25'], # ['qld', 'bm25', 'tct_colbert']
|
19 | 17 | 'metric': ['map'], # any valid trec_eval.9.0.4 metrics like ['map', 'ndcg', 'recip_rank']
|
|
0 commit comments