darpanaswal commited on
Commit
ba76ed7
·
verified ·
1 Parent(s): 3b98ef0

Update cross_encoder_reranking_train.py

Browse files
Files changed (1) hide show
  1. cross_encoder_reranking_train.py +1 -1
cross_encoder_reranking_train.py CHANGED
@@ -234,7 +234,7 @@ def main():
234
  parser.add_argument('--documents_content', type=str,
235
  default='./documents_content_with_features.json',
236
  help='Path to documents content JSON file')
237
- parser.add_argument('--queries_list', type=str, default='train_queries.json',
238
  help='Path to training queries JSON file')
239
  parser.add_argument('--text_type', type=str, default='TA',
240
  choices=['TA', 'claims', 'description', 'full', 'tac1', 'smart'],
 
234
  parser.add_argument('--documents_content', type=str,
235
  default='./documents_content_with_features.json',
236
  help='Path to documents content JSON file')
237
+ parser.add_argument('--queries_list', type=str, default='test_queries.json',
238
  help='Path to training queries JSON file')
239
  parser.add_argument('--text_type', type=str, default='TA',
240
  choices=['TA', 'claims', 'description', 'full', 'tac1', 'smart'],