darpanaswal commited on
Commit
29de29c
·
verified ·
1 Parent(s): f2bf451

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
@@ -214,7 +214,7 @@ def get_detailed_instruct(task_description: str, query: str) -> str:
214
  f"\nQuery: {query}"
215
  )
216
 
217
- def hybrid_score(cross_encoder_score, semantic_score, weight_cross=0.5, weight_semantic=0.5):
218
  return (weight_cross * cross_encoder_score) + (weight_semantic * semantic_score)
219
 
220
 
 
214
  f"\nQuery: {query}"
215
  )
216
 
217
+ def hybrid_score(cross_encoder_score, semantic_score, weight_cross=0.8, weight_semantic=0.2):
218
  return (weight_cross * cross_encoder_score) + (weight_semantic * semantic_score)
219
 
220