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 vote202 views1 answer
Pandas read_csv drop last column
Python
1 vote268 views1 answer
Max pooling tf keras
Python
0 vote217 views1 answer
Tf.slice
Python
0 vote219 views1 answer
Insert video in tkinter
Python
1 vote213 views1 answer
How to add a fuction in python
Python
4 votes215 views1 answer
Sparse categorical crossentropy
Python
22 votes220 views4 answers
Python count appearances in list
Python
5 votes189 views1 answer
Python read csv into array
Python
6 votes208 views2 answers
Multiple plot in one figure python
Python
0 vote194 views1 answer
Push notification using python
Python