numb3r3
commited on
Commit
·
c358563
1
Parent(s):
6e8f0b8
implement compute_score api
Browse files- modeling_bert.py +1 -0
modeling_bert.py
CHANGED
|
@@ -2119,6 +2119,7 @@ class JinaBertForSequenceClassification(JinaBertPreTrainedModel):
|
|
| 2119 |
)
|
| 2120 |
.float()
|
| 2121 |
)
|
|
|
|
| 2122 |
all_scores.extend(scores.cpu().numpy().tolist())
|
| 2123 |
|
| 2124 |
if len(all_scores) == 1:
|
|
|
|
| 2119 |
)
|
| 2120 |
.float()
|
| 2121 |
)
|
| 2122 |
+
scores = torch.sigmoid(scores)
|
| 2123 |
all_scores.extend(scores.cpu().numpy().tolist())
|
| 2124 |
|
| 2125 |
if len(all_scores) == 1:
|