From 6cfda0555cac808175598be5453017a67402cef8 Mon Sep 17 00:00:00 2001 From: Shahules786 Date: Thu, 25 Jan 2024 22:58:33 -0800 Subject: [PATCH] reshape matrix --- src/ragas/metrics/_answer_relevance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ragas/metrics/_answer_relevance.py b/src/ragas/metrics/_answer_relevance.py index 0c7795c03..3459530da 100644 --- a/src/ragas/metrics/_answer_relevance.py +++ b/src/ragas/metrics/_answer_relevance.py @@ -85,7 +85,7 @@ def calculate_similarity( question_vec = np.asarray(self.embeddings.embed_query(question)).reshape(1, -1) gen_question_vec = np.asarray( self.embeddings.embed_documents(generated_questions) - ) + ).reshape(len(generated_questions), -1) norm = np.linalg.norm(gen_question_vec, axis=1) * np.linalg.norm( question_vec, axis=1 )