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

0 vote118 views1 answer
Python catch int conversion error
Python
1 vote107 views1 answer
Split string every nth character
Python
7 votes121 views2 answers
How to execute python script in another script
Python
0 vote135 views1 answer
Check even or odd in single line
Python
0 vote122 views1 answer
Which type of programming does python support?
Python
8 votes117 views6 answers
Combine to lists python
Python
9 votes121 views1 answer
Comment out a block in python
Python
0 vote130 views1 answer
Flask new response style with `make_response`
Python Flask
0 vote133 views1 answer
Python how to automatically restart flask sever
Python
0 vote84 views1 answer
How to iterate a dictionary with minimum value in python
Python