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 vote354 views1 answer
Instal cython
Python
0 vote355 views1 answer
How to add phone number to django user model
Python
3 votes339 views1 answer
Month from datetime pandas
Python
0 vote335 views1 answer
Bytestring python
Python
5 votes403 views2 answers
Python Text Recognition
Python
1 vote343 views1 answer
Line continuation in python
Python
0 vote351 views3 answers
How to improve accuracy of random forest classifier
Python
1 vote333 views1 answer
Discord.py "NameError: name 'has_permissions' is not defined"
Python
0 vote350 views1 answer
Using django model translation with django rest
Python
0 vote366 views1 answer
How to give autocomplete in python
Python