Update base.py

This commit is contained in:
YangQianli92 2024-04-17 10:51:19 +08:00 committed by GitHub
parent b93e6779b9
commit bcc6691c4d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -148,6 +148,7 @@ class RAGBenchmark:
mrr = self.MRR(nodes, reference_doc)
similarity = await self.SemanticSimilarity(response, reference)
result = self._set_metrics(
bleu_avg, bleu1, bleu2, bleu3, bleu4, rouge_l,
similarity,