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 vote188 views1 answer
Torch.tensor.expand
Python
0 vote187 views1 answer
Python listas por comprension
Python
0 vote192 views1 answer
Filter a pandas dataframe by length of list in a column
Python
0 vote203 views1 answer
How do you make plot show with matplotlib ion method
Python
8 votes177 views5 answers
Python datetime weekday
Python
0 vote192 views1 answer
Python how to close the turtle tab on click
Python
3 votes183 views1 answer
How to make it so the pygame window will close
Python
0 vote190 views1 answer
Python opens windows store
Python
0 vote207 views1 answer
Autoscrapper import
Python
0 vote196 views1 answer
Python first two numbers
Python