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

19 votes159 views1 answer
Python ignore runtimewarning
Python
2 votes126 views3 answers
Condition python
Python
1 vote158 views1 answer
Last 24 hour python datetime
Python
0 vote133 views1 answer
Pandas group by multiple columns and count
Python
38 votes129 views2 answers
How to get the contents of a txt file in python
Python
0 vote130 views1 answer
How to get csv file first row first column value in python
Python
2 votes143 views3 answers
How to sum digits of a number in python
Python
7 votes125 views2 answers
What is the output of the following code? np.arange(0,22,6)
Python
3 votes123 views6 answers
Pandas count specific value in column
Python
0 vote133 views1 answer
Frozen graph can't be modified
Python