numb3r3
commited on
Commit
•
8283212
1
Parent(s):
8abb27c
implement compute_score api
Browse files- modeling_bert.py +1 -0
modeling_bert.py
CHANGED
|
@@ -2089,6 +2089,7 @@ class JinaBertForSequenceClassification(JinaBertPreTrainedModel):
|
|
| 2089 |
)
|
| 2090 |
.float()
|
| 2091 |
)
|
|
|
|
| 2092 |
all_scores.extend(scores.cpu().numpy().tolist())
|
| 2093 |
|
| 2094 |
if len(all_scores) == 1:
|
|
|
|
| 2089 |
)
|
| 2090 |
.float()
|
| 2091 |
)
|
| 2092 |
+
scores = torch.sigmoid(scores)
|
| 2093 |
all_scores.extend(scores.cpu().numpy().tolist())
|
| 2094 |
|
| 2095 |
if len(all_scores) == 1:
|