update sim test.

shibing624 · Jan 21, 2022 · ee34bd9 · ee34bd9
1 parent ba5d7cc
commit ee34bd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_similarity.py b/tests/test_similarity.py
@@ -76,7 +76,7 @@ def test_sbert_sim_batch(self):
         for i in range(len(sents1)):
             sims.append(scores[i][i])
         corr = compute_corrcoef(sims, labels)
-        print('scores:', scores[:10])
+        print('scores:', sims[:10])
         print('labels:', labels[:10])
         print('sbert_batch_sim spearman corr:', corr)