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 vote292 views2 answers
Binary Search tree implementation in python
Python
2 votes266 views1 answer
How to call super class constructor in python two classes
Python
4 votes279 views1 answer
Skip to next iteration in for loop python
Python
80 votes311 views6 answers
Sort dictionary by value python
Python
0 vote287 views1 answer
How to load keras model from json
Python
8 votes294 views1 answer
Tkinter clear entry
Python
7 votes276 views3 answers
How to run cmd command python
Python
0 vote297 views1 answer
Python check if hotkey pressed
Python
1 vote287 views3 answers
How to print to a file in python
Python
1 vote248 views1 answer
Create pyspark session with hive support
Python