Small fix for accuracy calculation for bert models

advaithsrao · Nov 26, 2023 · 38060b7 · 38060b7
1 parent 1cea7e8
commit 38060b7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/detector/modeler.py b/detector/modeler.py
@@ -187,7 +187,7 @@ def train(
                 # logits = logits.detach().cpu().numpy()
                 # label_ids = b_labels.detach().cpu().numpy()
 
-                total_eval_accuracy += self.accuracy(logits, label_ids)
+                total_eval_accuracy += self.accuracy(logits, b_labels)
 
             avg_val_accuracy = total_eval_accuracy / len(validation_dataloader)
             print(f'Validation Accuracy: {avg_val_accuracy:.4f}')
@@ -475,7 +475,7 @@ def train(
                 # logits = logits.detach().cpu().numpy()
                 # label_ids = b_labels.detach().cpu().numpy()
 
-                total_eval_accuracy += self.accuracy(logits, label_ids)
+                total_eval_accuracy += self.accuracy(logits, b_labels)
 
             avg_val_accuracy = total_eval_accuracy / len(validation_dataloader)
             print(f'Validation Accuracy: {avg_val_accuracy:.4f}')