Update evaluator.py

jhaayush2004 · Aug 11, 2024 · 4e6d7e6 · 4e6d7e6
1 parent 8f9a3b2
commit 4e6d7e6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rag-evaluator/evaluator.py b/rag-evaluator/evaluator.py
@@ -8,7 +8,8 @@ def __init__(self, model_name='paraphrase-MiniLM-L6-v2', top_n_keywords=50):
         self.model = SentenceTransformer(model_name)
         self.tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
         self.top_n_keywords = top_n_keywords
-
+
+    '''Defining evaluate_semantic_similarity'''
     def evaluate_semantic_similarity(self, reference_text, generated_text):
         reference_embedding = self.model.encode(reference_text).reshape(1, -1)
         generated_embedding = self.model.encode(generated_text).reshape(1, -1)