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

5 votes217 views1 answer
Q is not defined pylance django
Python
0 vote268 views1 answer
Form is undefined flask
Python Flask
4 votes238 views3 answers
Display prime numbers between two intervals in python
Python
0 vote237 views1 answer
Matplotlib pyplot comment on plot
Python
0 vote232 views1 answer
Who is bayceee roblox id
Python
4 votes232 views2 answers
Median of a list python
Python
12 votes232 views2 answers
Filtering a list
Python
0 vote273 views3 answers
Django forms I cannot save picture file
Python Django
-1 vote244 views1 answer
Magic line not found jupyter notebook
Python
3 votes270 views1 answer
Pandas plotly
Python