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 vote106 views4 answers
Regex find all french phone number python
Python
3 votes124 views2 answers
Python obfuscator
Python
0 vote137 views1 answer
Foreign key django createview
Python Django
0 vote123 views3 answers
How to draw dendrogram in python
Python
0 vote124 views1 answer
Hackerrank python solutions
Python
0 vote102 views1 answer
Shared SHMEM python
Python
11 votes121 views2 answers
Python execute time
Python
13 votes112 views3 answers
Np.hstack
Python
0 vote164 views1 answer
Discord.py how to print audit logs
Python
0 vote107 views1 answer
Is python platform independent
Python