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 vote324 views1 answer
Django charfield force lowercase
Python
-1 vote312 views1 answer
Why women are better than men
Python
0 vote326 views1 answer
Tessa thompson
Python
1 vote314 views1 answer
Split a given number in python
Python
7 votes357 views2 answers
Mongodb aggregate count
Python
0 vote303 views3 answers
Discord.py delete own message
Python
0 vote317 views1 answer
How to keep old content when using write in python
Python
0 vote325 views1 answer
PIL Make Circle
Python
0 vote304 views1 answer
Bulk file name changer in python
Python
6 votes368 views3 answers
Cosine similarity python
Python