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 vote214 views1 answer
Python code for system of odes
Python
20 votes198 views2 answers
Python how to count the lines in a file
Python
15 votes203 views1 answer
Python add to list with index
Python
2 votes179 views1 answer
Private variables python
Python
1 vote190 views1 answer
All pdf in a directory to csv python
Python
3 votes232 views6 answers
2+2
Python
0 vote177 views1 answer
Boolean meaning in python
Python
1 vote168 views1 answer
Print value of array python
Python
1 vote178 views1 answer
Evaluate how much a python program memory
Python
0 vote247 views1 answer
Day 27: Testing hackerrank solution python
Python