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

1 vote98 views1 answer
Print upto 1 decimal place python
Python
0 vote90 views1 answer
Stack in python using linked list
Python
0 vote117 views1 answer
Sending email in django
Python Django
0 vote106 views3 answers
2 decimal places
Python
0 vote145 views1 answer
Advanced use of tuples in python
Python
0 vote97 views2 answers
Python function overloading
Python
0 vote100 views1 answer
Replace nan in pandas column with mode and printing it
Python
0 vote87 views1 answer
Flask No application found. Either work inside a view function or push an application context
Python
5 votes93 views1 answer
Mp4 get all images frame by frame python
Python
0 vote100 views1 answer
Find no of iterations in python
Python