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

0 vote221 views1 answer
Python regular expression escape vertical bar |
Python
0 vote180 views2 answers
Copy only some columns to new dataframe in r
Python
4 votes147 views2 answers
Finding median on python
Python
0 vote147 views1 answer
Pandas concat / merge two dataframe within one dataframe
Python
8 votes139 views2 answers
Python get base directory
Python
0 vote136 views1 answer
Arabert
Python
0 vote168 views1 answer
Creer des disques en python tkinter
Python
13 votes154 views3 answers
Python http server command line
Python
0 vote155 views1 answer
Typecasting inpython
Python
0 vote167 views2 answers
Find location of a class in python
Python