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 vote330 views1 answer
Couchbase python
Python
0 vote314 views1 answer
Dont truncate dataframe jupyter pd display options
Python
1 vote314 views3 answers
How to access variable of one function in another function in python
Python
2 votes331 views1 answer
How to load ui file in pyqt5
Python
0 vote364 views1 answer
% operatior in python print
Python
0 vote363 views1 answer
Zoom in geopandas polot
Python
10 votes330 views1 answer
Python check if number is float or int
Python
9 votes317 views6 answers
Join pandas dataframe by column
Python
0 vote338 views1 answer
Don't get volunteered foobar
Python
0 vote322 views2 answers
Open file dialog on button click pyqt5
Python