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

6 votes258 views2 answers
Delete variable python
Python
0 vote287 views1 answer
Python discord know message from bot
Python
120 votes260 views6 answers
Python random number
Python
16 votes283 views1 answer
Python reload file if changed
Python
8 votes265 views6 answers
Calculator in python
Python
0 vote252 views1 answer
How to replace an element of a list using list comprehension
Python
0 vote258 views2 answers
Set cookie in python requests
Python
5 votes256 views4 answers
Python get last element in dict
Python
12 votes244 views3 answers
Delete all files in a directory python
Python
23 votes387 views3 answers
Django rest framework
Python Django