()
| 137 | |
| 138 | |
| 139 | def get_args(): |
| 140 | parser = argparse.ArgumentParser() |
| 141 | parser.add_argument("--index_dir_path", type = str, default="/media/nvme/fengran/index/bm25_topiocqa") |
| 142 | parser.add_argument("--query_file", type = str, default="/media/nvme/fengran/output/combine_topiocqa/mistral_TopiOCQA_test_AD+FT.json", nargs='+') |
| 143 | parser.add_argument("--gold_qrel_file_path", type=str, default="/media/nvme/fengran/TopiOCQA/topiocqa_qrel.tsv") |
| 144 | parser.add_argument("--not_perform_evaluation", action="store_true", default=False) |
| 145 | parser.add_argument("--top_n", type=int, default=100) |
| 146 | parser.add_argument("--bm25_k1", type=float, default=0.9) # 0.82 for qrecc, 0.9 for topiocqa |
| 147 | parser.add_argument("--bm25_b", type=float, default=0.4) # 0.68 for qrecc, 0.4 for topiocqa |
| 148 | parser.add_argument("--rel_threshold", type=int, default=1) |
| 149 | parser.add_argument("--split_num_chunk", type=int, default=1) |
| 150 | parser.add_argument("--max_seq_length", type=int, default=512) |
| 151 | parser.add_argument("--retrieval_output_path", type=str, default="data/topiocqa/qrel/bm25") |
| 152 | parser.add_argument("--output_file_name", type=str, default="mistral_TopiOCQA_test_AD+FT.trec", nargs='+') |
| 153 | args = parser.parse_args() |
| 154 | |
| 155 | logger.info("---------------------The arguments are:---------------------") |
| 156 | pprint(args) |
| 157 | assert len(args.query_file) == len(args.output_file_name) |
| 158 | return args |
| 159 | |
| 160 | if __name__ == '__main__': |
| 161 | args = get_args() |
no outgoing calls
no test coverage detected