Update cross_encoder_reranking_train.py

cross_encoder_reranking_train.py

@@ -214,7 +214,7 @@ def get_detailed_instruct(task_description: str, query: str) -> str:
         f"\nQuery: {query}"
     )
 
-def hybrid_score(cross_encoder_score, semantic_score, weight_cross=0.7, weight_semantic=0.3):
+def hybrid_score(cross_encoder_score, semantic_score, weight_cross=0.5, weight_semantic=0.5):
     return (weight_cross * cross_encoder_score) + (weight_semantic * semantic_score)
 
 
@@ -327,7 +327,7 @@ def main():
                         default='./documents_content_with_features.json',
                         help='Path to documents content JSON file')
     # Change here from train to test
-    parser.add_argument('--queries_list', type=str, default='train_queries.json',
+    parser.add_argument('--queries_list', type=str, default='test_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'],