- from semantic_split import SentenceTransformersSimilarity
- def test_basic_usage():
- sentence0 = "I love dogs."
- sentence1 = "I love cats."
- model = SentenceTransformersSimilarity()
- similarities = model.similarities([sentence0, sentence1])
- assert similarities[0] - 0.7728430 < 0.0001
|