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 vote166 views1 answer
Python condition if dataype
Python
1 vote171 views1 answer
How to add three conditions in np.where in pandas dataframe
Python
-1 vote165 views2 answers
Grab the first letter of each string in an array python
Python
15 votes157 views1 answer
Format to 2 or n decimal places python
Python
0 vote159 views1 answer
Split dataset into train, test and validation sets
Python
8 votes164 views1 answer
Jupyter column limit pd
Python
1 vote175 views1 answer
Train,test,dev python
Python
0 vote171 views1 answer
Got AttributeError when attempting to get a value for field `name` on serializer
Python
4 votes121 views1 answer
Should i learn python or javascript
Python
29 votes211 views6 answers
Create empty pandas dataframe
Python