fix bugs in cosine_match.py

fix bug
TablewareBox · Mar 11, 2024 · 8ac8c78 · 8ac8c78
1 parent 5cd4d3e
commit 8ac8c78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evals/elsuite/cosine_match.py b/evals/elsuite/cosine_match.py
@@ -334,7 +334,7 @@ def eval_sample(self, test_sample, rng):
         )
         evals.record.record_metrics(
             value_recall = sum(matches)/len(matches) if matches else 0.0,
-            f1_score = utils.macro_f1_score_3(self.word2vec_model, sample_list, correct_answers),
+            f1_score = utils.cos_f1_score(self.word2vec_model, sample_list, correct_answers),
         )
 
     def run(self, recorder: RecorderBase):