浏览代码

fix issue with division by 0

Elena Paraschiv 9 月之前
父节点
当前提交
d2a3ca8174
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      llmeval/commons/management/commands/eval_qa.py

+ 2 - 2
llmeval/commons/management/commands/eval_qa.py

@@ -362,7 +362,7 @@ class Command(BaseCommand):
             eval_answer.save()
             eval_answer.save()
             if session.request_delay:
             if session.request_delay:
                 time.sleep(session.request_delay)
                 time.sleep(session.request_delay)
-
-        print(f"Accuracy: {stats['correct']}/{stats['total']} ({stats['correct']/stats['total']:.2f})")
+        if stats['total']>0:
+            print(f"Accuracy: {stats['correct']}/{stats['total']} ({stats['correct']/stats['total']:.2f})")
 
 
         self.stdout.write(self.style.SUCCESS('Successfully evaluated MedQA'))
         self.stdout.write(self.style.SUCCESS('Successfully evaluated MedQA'))