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 vote310 views2 answers
Matplotlib display graph on jupyter notebook
Python
0 vote353 views2 answers
Cronometer python
Python
31 votes326 views6 answers
Upgrade python version
Python
0 vote315 views1 answer
Count how many loops that printed in python
Python
0 vote316 views1 answer
Python sum of 10 numbers from user input
Python
71 votes353 views6 answers
Python ide online
Python
0 vote317 views1 answer
Pandas use map lambda to fillna python
Python
68 votes337 views5 answers
Python main method
Python
0 vote342 views1 answer
Django graphene without model
Python Django
0 vote339 views2 answers
Spyder 32 bit download
Python