Spaces:

darpanaswal
/

Patent_Retrieval

Configuration error

darpanaswal commited on Apr 11

Commit

f7855d2

verified ·

1 Parent(s): fb46599

Update cross_encoder_reranking_train.py

Files changed (1) hide show

cross_encoder_reranking_train.py CHANGED Viewed

@@ -32,7 +32,7 @@ def rank_by_centrality(texts):
     ranked = sorted(zip(texts, centrality_scores), key=lambda x: x[1], reverse=True)
     return [text for text, _ in ranked]
-def cluster_and_rank(texts, threshold=0.8):
     if len(texts) < 2:
         return texts

     ranked = sorted(zip(texts, centrality_scores), key=lambda x: x[1], reverse=True)
     return [text for text, _ in ranked]
+def cluster_and_rank(texts, threshold=0.6):
     if len(texts) < 2:
         return texts