dzenzzz commited on
Commit
324ef5b
·
1 Parent(s): d3d68bb

minor change

Browse files
Files changed (1) hide show
  1. 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]