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 vote130 views1 answer
Django genericforeignkey null
Python
0 vote135 views1 answer
Python extend vs append
Python
1 vote142 views1 answer
Create period pandas
Python
0 vote233 views1 answer
Numpy randn with a shape of another array
Python
5 votes132 views1 answer
How to round to 2 decimals with Python?
Python
1 vote133 views1 answer
Torch save state dict
Python
0 vote139 views9 answers
Odoo.py odoo 14
Python
1 vote159 views1 answer
Pip uninstalled itself
Python
2 votes123 views1 answer
Check if part of list is in another list python
Python
0 vote146 views1 answer
Fetch email from gmail using python site:stackoverflow.com
Python