Spaces:
Sleeping
Sleeping
minor change
Browse files- reranker.py +1 -0
reranker.py
CHANGED
|
@@ -53,6 +53,7 @@ class Reranker:
|
|
| 53 |
batch_scores = torch.nn.functional.log_softmax(batch_scores, dim=1)
|
| 54 |
scores = batch_scores[:, 1].exp().tolist()
|
| 55 |
|
|
|
|
| 56 |
results = list(zip(scores, documents))
|
| 57 |
results.sort(key=lambda x: x[0], reverse=True)
|
| 58 |
top_10 = results[:10]
|
|
|
|
| 53 |
batch_scores = torch.nn.functional.log_softmax(batch_scores, dim=1)
|
| 54 |
scores = batch_scores[:, 1].exp().tolist()
|
| 55 |
|
| 56 |
+
|
| 57 |
results = list(zip(scores, documents))
|
| 58 |
results.sort(key=lambda x: x[0], reverse=True)
|
| 59 |
top_10 = results[:10]
|