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

3 votes149 views1 answer
Best games made in pygame
Python
0 vote180 views1 answer
Supervisor gunicorn virtualenv flask
Python Flask
0 vote151 views1 answer
Python von konsoleeinlesen
Python
1 vote170 views1 answer
Requirements.txt flask
Python
2 votes154 views1 answer
Show rows with a null value pandas
Python
-1 vote144 views1 answer
Returns a new DataFrame containing the distinct rows in this DataFrame
Python
5 votes189 views3 answers
Flask port
Python Flask
4 votes130 views4 answers
Python save output to file
Python
2 votes137 views2 answers
Python django query
Python
0 vote132 views1 answer
Get a column of a csv python
Python