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

23 votes99 views6 answers
Python even or odd
Python
0 vote142 views2 answers
Count items in a model django rest
Python Django
0 vote133 views1 answer
Pandas Columns Calling using read_csv
Python
0 vote144 views1 answer
Settings.debug django
Python Django
0 vote133 views1 answer
Read variable in a string python
Python
1 vote102 views1 answer
Python 3d array
Python
0 vote112 views1 answer
Start ipython with any version
Python
9 votes116 views2 answers
Read video with opencv
Python
0 vote108 views1 answer
Operation that returns True if all values are equal
Python
4 votes97 views1 answer
Pandas dataframe replace inf
Python