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

3 votes137 views1 answer
Matplotlib text too small
Python
4 votes112 views1 answer
Python today - 1 day
Python
1 vote125 views1 answer
Django loop index
Python Django
-1 vote110 views2 answers
Find common values in different dataframes pandas
Python
1 vote132 views2 answers
Django password hashing
Python
1 vote134 views1 answer
How to count post by category django
Python Django
5 votes107 views1 answer
Python how move file to directory
Python
7 votes111 views3 answers
How to run cmd command python
Python
0 vote107 views2 answers
Pandas delete first row
Python
0 vote106 views1 answer
Python turn list of strings into list of doubles
Python