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 vote133 views1 answer
Fill python list with input
Python
0 vote140 views2 answers
How to convert c to python
Python
0 vote82 views0 answer
Python
46 votes140 views6 answers
Python for loop array index
Python
0 vote121 views1 answer
How to save date in cookie Python
Python
0 vote137 views1 answer
Dynamic array python numpy
Python
0 vote124 views1 answer
What is {:02d}:{:02d} in python
Python
3 votes171 views2 answers
Publisher python ros
Python
2 votes118 views1 answer
Python get all methods of object
Python
0 vote132 views1 answer
Pandas split groupby
Python