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 vote201 views1 answer
Python3 strip punctuation from string
Python
0 vote248 views1 answer
ModuleNotFoundError: No module named 'readline'
Python
14 votes208 views4 answers
Python pip jupyter notebook install
Python
86 votes191 views3 answers
Install python 3.8 linux
Python
0 vote214 views1 answer
Install os conda
Python
0 vote205 views1 answer
Python multiline code dot
Python
1 vote195 views1 answer
Tkinter new line in text
Python
0 vote195 views1 answer
Prime number checking algorithm
Python
1 vote196 views1 answer
How to pip install pyttsx3
Python
0 vote195 views1 answer
Shorter Max fonction code in python
Python