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

9 votes148 views2 answers
Pygame render text
Python
0 vote153 views1 answer
Python path zsh mac
Python
0 vote148 views1 answer
E: Package 'python-certbot-nginx' has no installation candidate
Python
14 votes156 views3 answers
Pillow image from array
Python
0 vote141 views1 answer
MIn-Max problem in python
Python
7 votes195 views2 answers
Show image in python
Python
1 vote189 views1 answer
Plt.plot width line
Python
1 vote217 views1 answer
Curso python
Python
0 vote170 views1 answer
Pygame zero how to draw text
Python
0 vote177 views1 answer
Nltk stop words
Python