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 vote158 views1 answer
Concatenar columnas en una del mismo dataset
Python
1 vote164 views1 answer
Install python packages from inside within python program
Python
2 votes162 views4 answers
How to read second sheets of excel using python
Python
3 votes160 views1 answer
Python docker
Python
0 vote160 views1 answer
Django check user admin
Python
6 votes153 views1 answer
How to simulate a key press in python
Python
0 vote154 views1 answer
Hello world in dip
Python
8 votes158 views2 answers
Matplotlib x label rotation
Python
9 votes162 views4 answers
Scikit learn random forest classifier
Python
0 vote159 views1 answer
Numpy substract subsequent elements
Python