jinaai
/

jina-reranker-v1-tiny-en

Feature Extraction

Transformers.js

🇪🇺 Region: EU

Model card Files Files and versions Community

numb3r3 commited on Apr 16

Commit

c358563

•

1 Parent(s): 6e8f0b8

implement compute_score api

Files changed (1) hide show

modeling_bert.py +1 -0

modeling_bert.py CHANGED Viewed

@@ -2119,6 +2119,7 @@ class JinaBertForSequenceClassification(JinaBertPreTrainedModel):
                 )
                 .float()
             )
             all_scores.extend(scores.cpu().numpy().tolist())
         if len(all_scores) == 1:

                 )
                 .float()
             )
+            scores = torch.sigmoid(scores)
             all_scores.extend(scores.cpu().numpy().tolist())
         if len(all_scores) == 1: