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 vote286 views1 answer
Pafy documentation
Python
4 votes326 views3 answers
How to duplicate a list in python
Python
0 vote371 views1 answer
Python window icon on task bar
Python
0 vote330 views1 answer
Nornir config file
Python
0 vote338 views1 answer
Django command to fetch all columns of a table
Python
0 vote315 views1 answer
Django.db.backends.mysql install
Python
0 vote317 views2 answers
How to add space in python
Python
0 vote326 views2 answers
Python object name
Python
0 vote344 views1 answer
Drop first column read_csv
Python
1 vote309 views1 answer
Pygame how to get surface lenght
Python