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

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

def calculate_answer_similarity(user_answer, expected_answer):
    emb1 = model.encode(user_answer, convert_to_tensor=True)
    emb2 = model.encode(expected_answer, convert_to_tensor=True)

    score = util.pytorch_cos_sim(emb1, emb2)

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