test_sentence_similarity.py 306 B

12345678910111213
  1. from semantic_split import SentenceTransformersSimilarity
  2. def test_basic_usage():
  3. sentence0 = "I love dogs."
  4. sentence1 = "I love cats."
  5. model = SentenceTransformersSimilarity()
  6. similarities = model.similarities([sentence0, sentence1])
  7. assert similarities[0] - 0.7728430 < 0.0001