File size: 360 Bytes
438c749
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from sentence_transformers import SentenceTransformer, util

model = SentenceTransformer("all-MiniLM-L6-v2")

def topic_relevance(user_answer, topic):

    emb1 = model.encode(user_answer, convert_to_tensor=True)
    emb2 = model.encode(topic, convert_to_tensor=True)

    score = util.pytorch_cos_sim(emb1, emb2)

    return round(float(score[0][0]) * 100, 2)