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

0 vote131 views1 answer
Check processing bar of loop in python
Python
14 votes139 views6 answers
Django install
Python
0 vote137 views1 answer
Dictionary default value python if key not found
Python
0 vote184 views1 answer
Django filter word count greater than
Python Django
0 vote131 views1 answer
Numpy transpose shorthand
Python
72 votes130 views6 answers
Python loop array
Python
0 vote138 views1 answer
Make image to string to use in tkinter
Python
4 votes162 views1 answer
Import sklearn.metrics from plot_confusion_matrix
Python
3 votes170 views2 answers
Python extract specific columns from pandas dataframe
Python
62 votes136 views6 answers
Class python
Python