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

10 votes262 views3 answers
Matplotlib axis labels
Python
39 votes263 views6 answers
Convert dict to dataframe
Python
0 vote292 views1 answer
Seaborn heatmap center xticks
Python
0 vote287 views1 answer
Pandas df where row has na
Python
0 vote299 views1 answer
Sklearn impute
Python
0 vote296 views1 answer
How to estimate memory of dataset using python command
Python
12 votes301 views2 answers
Python input integer only
Python
0 vote303 views3 answers
Hash function in Python
Python
0 vote273 views1 answer
File = Root() path = file.fileDialog() print("PATH = ", path)
Python
2 votes301 views2 answers
How to convert tensor to list tensorflow
Python