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 vote181 views1 answer
Keep the user logged in even though user changes password django
Python
7 votes230 views2 answers
From distutils.util import strtobool ModuleNotFoundError: No module named 'distutils.util'
Python
0 vote174 views2 answers
Error: Could not locate a Flask application. You did not provide the "FLASK_APP" environment variable, and a "wsgi.py" or "app.py" module was not found in the current directory.
Python
0 vote174 views1 answer
Class Student(): def name(self):
Python
0 vote252 views1 answer
How to select the shortest item in a python list
Python
133 votes166 views6 answers
Python input
Python
0 vote210 views1 answer
Pandas Columns Calling using read_csv
Python
0 vote194 views1 answer
Numpy multidimensional indexing
Python
0 vote191 views1 answer
Python regex find single character
Python
1 vote168 views2 answers
Discord py get channel ID by name
Python