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 vote225 views4 answers
Flask activ environment
Python Flask
0 vote244 views1 answer
How to insert value in admin panel in django
Python Django
0 vote171 views1 answer
Subprocess the system cannot find the file specifie
Python
5 votes224 views2 answers
How to draw image in tkinter
Python
2 votes166 views1 answer
Python random date between range
Python
0 vote165 views1 answer
Grepper how to use fraction
Python
0 vote151 views1 answer
Create django model field based on another field
Python
0 vote228 views1 answer
Binarizer pyspark
Python
-86 votes159 views10 answers
Code
Python
2 votes160 views2 answers
Python add list to dictionary in loop
Python