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 votes179 views2 answers
Python change cwd to script directory
Python
1 vote167 views1 answer
Groupby year datetime pandas
Python
0 vote179 views1 answer
Df.min python
Python
2 votes181 views1 answer
Python check tuple length
Python
0 vote137 views1 answer
Python fft frequency spectrum
Python
0 vote218 views5 answers
Convert c++ code to python
Python
0 vote169 views1 answer
How to open camre aopencv
Python
33 votes161 views4 answers
Change dataframe column type
Python
3 votes246 views6 answers
Get multiple inputs in python
Python
0 vote169 views1 answer
Python print without optional argument
Python