Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -313,7 +313,7 @@ def main():
     parser = argparse.ArgumentParser(description='Re-rank patents using cross-encoder scoring (training queries only)')
     parser.add_argument('--pre_ranking', type=str, default='shuffled_pre_ranking.json',
                         help='Path to pre-ranking JSON file')
-    parser.add_argument('--output', type=str, default='predictions2.json',
+    parser.add_argument('--output', type=str, default='prediction2.json',
                         help='Path to output re-ranked JSON file')
     parser.add_argument('--queries_content', type=str, 
                         default='./queries_content_with_features.json',
@@ -321,7 +321,8 @@ def main():
     parser.add_argument('--documents_content', type=str, 
                         default='./documents_content_with_features.json',
                         help='Path to documents content JSON file')
-    parser.add_argument('--queries_list', type=str, default='test_queries.json',
+    # Change here from train to test
+    parser.add_argument('--queries_list', type=str, default='train_queries.json',
                         help='Path to training queries JSON file')
     parser.add_argument('--text_type', type=str, default='TA',
                         choices=['TA', 'claims', 'description', 'full', 'tac1', 'smart', 'smart2'],