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 vote217 views1 answer
Username and Password Login Function
Python
1 vote259 views2 answers
Python print an array
Python
10 votes247 views6 answers
Flask – Environment
Python Flask
0 vote207 views1 answer
Numpy diff
Python
8 votes212 views3 answers
Normalize data python
Python
0 vote210 views1 answer
Python modules screen
Python
0 vote151 views1 answer
Shutil copy file python
Python
2 votes202 views1 answer
Discord.py add role on member join
Python
1 vote213 views2 answers
Rename files in a folder python
Python
3 votes199 views1 answer
How to open webcam with python
Python