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

-4 votes300 views4 answers
Latest django version
Python
0 vote310 views2 answers
Python convert array to linked list
Python
3 votes290 views1 answer
Python check array exists
Python
3 votes269 views2 answers
How to create a label in python
Python
2 votes278 views2 answers
Python replace multiple spacis with spaces
Python
1 vote288 views1 answer
How to set required drf serialzier
Python
0 vote289 views1 answer
Passport parsing python
Python
4 votes291 views2 answers
Endswith python
Python
7 votes301 views1 answer
Check key pressed pygame
Python
19 votes286 views3 answers
Convert a list ['Pythons syntax is easy to learn', 'Pythons syntax is very clear'] to a string using ‘&’.
Python