naver
/

xprovence-reranker-bgem3-v1

feature-extraction

Model card Files Files and versions

youssef101 commited on Sep 4

Commit

fae7fc0

·

verified ·

1 Parent(s): 52a2c44

Update modeling_xprovence_hf.py

Files changed (1) hide show

modeling_xprovence_hf.py +1 -1

modeling_xprovence_hf.py CHANGED Viewed

@@ -133,7 +133,7 @@ class XProvence(XLMRobertaPreTrainedModel):
         context: Union[List[List[str]], str],
         title: Optional[Union[List[List[str]], str]] = "first_sentence",
         batch_size=32,
-        threshold=0.1,
         always_select_title=False,
         reorder=False,
         top_k=5,

         context: Union[List[List[str]], str],
         title: Optional[Union[List[List[str]], str]] = "first_sentence",
         batch_size=32,
+        threshold=0.3,
         always_select_title=False,
         reorder=False,
         top_k=5,