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 vote233 views2 answers
Pygame keys keep pressing
Python
0 vote286 views1 answer
Django check if user is staff in template
Python Django
-1 vote235 views1 answer
Colours seaborn
Python
12 votes246 views10 answers
Fortnite
Python
0 vote244 views2 answers
Headers cors
Python
6 votes217 views2 answers
Python max
Python
1 vote245 views2 answers
Linspace python
Python
0 vote226 views1 answer
57 *2
Python
0 vote263 views1 answer
Python get screen size raspberry pi
Python
1 vote243 views1 answer
Pandas query variable count
Python