Sentence similarity python

# sentence_transformer: Python module using Sentence BERT 
# check documentation in source link for further details
from sentence_transformers import SentenceTransformer
from sentence_transformers.util import pytorch_cos_sim # cosine silarity

# stsb-roberta-large is one of the possible pre-trained models
model = SentenceTransformer('stsb-roberta-large')

# list of sentences (type: str)
sentences = list(...) 

# calculate the embeddings
embeddings = bert_model.encode(sentences)

# example: cosine similarity among first and second sentences
cos_similarity = pytorch_cos_sim(embeddings[0], embeddings[1])

Source: sbert.net

Similar codes

37 votes186 views1 answer
Python make txt file
Python
3 votes220 views2 answers
Variance calculation python manually
Python
0 vote182 views1 answer
Add Cog to bot in Discord.py
Python
1 vote193 views1 answer
Object.image.url email template django
Python
2 votes195 views2 answers
Find prime number in given range in python
Python
5 votes179 views6 answers
How to check a string in if statement python
Python
1 vote191 views2 answers
Python line break inside string
Python
0 vote254 views1 answer
No module named 'bootstrap4' django
Python Django
1 vote194 views1 answer
Guido van rossum net worth
Python
0 vote213 views1 answer
Python screeninfo
Python