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

222 votes435 views6 answers
Python random numbers
Python
0 vote148 views1 answer
Flip a coin with array in python
Python
2 votes137 views2 answers
Python infinite l00p
Python
0 vote177 views1 answer
Django not detecting new model
Python Django
0 vote149 views1 answer
Python 2.7 get user input
Python
1 vote190 views1 answer
Reached 'max' / getOption("max.print")
Python
0 vote153 views1 answer
Django template child data in nested loop
Python
0 vote149 views1 answer
Generate a random letter using python
Python
0 vote157 views2 answers
Python how to keep turtle window open
Python
19 votes166 views2 answers
Learn python 3
Python