Variable is initialized once

new_test_saved_finetuned_model.py

@@ -179,7 +179,8 @@ class BERTFineTuneTrainer:
                 # true_labels = torch.argmax(data["label"], dim=-1)
                 plabels.extend(predicted_labels.cpu().numpy())
                 tlabels.extend(data['label'].cpu().numpy())
-                positive_class_probs = [prob[1] for prob in probabs]
+                # positive_class_probs = [prob[1] for prob in probabs]
+                positive_class_probs.append(probs.detach().cpu().numpy().tolist()[1])
                 # Compare predicted labels to true labels and calculate accuracy
                 correct = (data['label'] == predicted_labels).sum().item()