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 vote176 views1 answer
Count missing values groupby
Python
4 votes194 views2 answers
Freq count in python
Python
6 votes222 views3 answers
Pandas to csv without header
Python
1 vote208 views1 answer
How to make a kivy label multiline text
Python
61 votes207 views6 answers
Python sort dict by key
Python
0 vote180 views0 answer
Python
15 votes181 views10 answers
Hello world
Python
0 vote201 views1 answer
Renaming Column Name Dataframe
Python
1 vote184 views1 answer
Data Compression in python
Python
5 votes252 views2 answers
Zipfile python unzip with path
Python