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 vote203 views1 answer
Python input character limit
Python
43 votes280 views2 answers
For loop in df rows
Python
0 vote209 views1 answer
Python read column from csv
Python
0 vote205 views1 answer
How to change a particular text inside a list of dictionary
Python
32 votes227 views6 answers
Remove space from string python
Python
1 vote200 views1 answer
Placeholder tkinter
Python
0 vote257 views1 answer
Python conda how to see channels command
Python
1 vote215 views1 answer
USB: usb_device_handle_win.cc:1049 Failed to read descriptor from node connection: A device attached to the system is not functioning. (0x1F)
Python
3 votes198 views2 answers
Sort tuple by first element python
Python
0 vote241 views1 answer
Fast comand exit python windows
Python