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

19 votes172 views4 answers
Install python on ubuntu
Python
0 vote191 views1 answer
How to check if given primary key exists in django model
Python
1 vote173 views1 answer
Server on python
Python
8 votes183 views2 answers
Pandas filter rows by value in list
Python
2 votes173 views1 answer
Django override delete
Python
1 vote160 views1 answer
Python append several values
Python
18 votes169 views5 answers
Get list with random numbers python
Python
1 vote177 views1 answer
Python region
Python
10 votes186 views1 answer
Python turtle
Python
1 vote197 views1 answer
Print list vertically in python with loop
Python