adding missing function

place_gen_evaluate.py

@@ -81,6 +81,24 @@ class Place_gen_evaluate(evaluate.Metric):
         self.bleu = evaluate.load('bleu')
         self.bert_score = evaluate.load('bertscore')
 
+    def _calculate_pre_rec(self, gens, golds, d):
+        dists = []
+    
+        for gold in golds:
+            g_dist = []
+            for gen in gens:
+                g_dist.append(d(gold, gen))
+            if len(g_dist) == 0:
+                g_dist.append(100) #penalty if the model doesnt generate anything
+            dists.append(g_dist)
+    
+        dists = np.array(dists)
+        precision = np.min(dists, axis=0).sum()
+    
+        recall = np.min(dists, axis=1).sum()
+    
+        return precision, recall
+
     def _compute(self, generations, golds):
         '''Calculate Accuracy and BLEU-1 scores between model generations and golden answers.
            We expect a set of generated answers and want to find it among a set of gold answers.'''
@@ -106,7 +124,7 @@ class Place_gen_evaluate(evaluate.Metric):
             f_ans = list(set([a.lower().strip() for a in f_ans])) #get rid of duples values
             f_gold = list(set(gold))
     
-            precision, recall = calculate_pre_rec(f_ans, f_gold, Levenshtein.distance)
+            precision, recall = self._calculate_pre_rec(f_ans, f_gold, Levenshtein.distance)
     
             precisions.append(precision)
             recalls.append(recall)