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 vote259 views1 answer
Unlimited keyword arguments python
Python
1 vote256 views1 answer
How to get images on flask page
Python
42 votes243 views4 answers
Replace column values pandas
Python
0 vote258 views1 answer
Equivalenci EN PYTHON DE INPUT EN C#
Python
1 vote256 views1 answer
How to split a string by space in python
Python
1 vote252 views1 answer
R vs python
Python
4 votes249 views2 answers
Is python easy or hard to learn
Python
4 votes285 views6 answers
Function in function python
Python
9 votes261 views3 answers
Print current time hours and minutes in python
Python
12 votes239 views6 answers
Python remove empty string from list
Python