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

1 vote282 views1 answer
Convert list of values to dictionary based on count
Python
10 votes262 views4 answers
Python multiply all elements in list
Python
6 votes269 views6 answers
What is the best ide for python
Python
0 vote270 views1 answer
A int and float. python
Python
0 vote304 views1 answer
Python - Modify Strings
Python
0 vote285 views1 answer
Créer csv python
Python
-1 vote254 views1 answer
Python Sort Lists
Python
1 vote277 views1 answer
Python delete text in text file
Python
0 vote279 views1 answer
Python how to automatically restart flask sever
Python
0 vote235 views1 answer
Count occurrences of one variable grouped by another python
Python