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 vote218 views1 answer
Calculate the euler totient of a number python
Python
0 vote260 views1 answer
How to send image to template thats not in static flask
Python Flask
43 votes206 views6 answers
Today date in python
Python
4 votes234 views1 answer
Convert string array to integer python
Python
0 vote266 views1 answer
How i rwrite conditon to create 1 or 0 label from two probability column python
Python
0 vote227 views1 answer
Discord bot slash
Python
0 vote207 views1 answer
Search bar using python
Python
1 vote198 views1 answer
Python tkinter change color of main window
Python
0 vote219 views2 answers
Django create view filter options
Python
1 vote213 views1 answer
Filter dataframe columns vy a list of columns
Python