Logs

app.py

@@ -199,6 +199,10 @@ dataloader = DataLoader(tokenized_data['train'], batch_size=batch_size, collate_
 # create testing data loader
 test_dataloader = DataLoader(tokenized_data['test'], batch_size=batch_size, collate_fn=data_collator)
 
+# TEMPORARY - for testing
+teacher_accuracy, teacher_precision, teacher_recall, teacher_f1 = evaluate_model(teacher_model, test_dataloader, device)
+print(f"Teacher (test) - Accuracy: {teacher_accuracy:.4f}, Precision: {teacher_precision:.4f}, Recall: {teacher_recall:.4f}, F1 Score: {teacher_f1:.4f}")
+
 # put student model in train mode
 student_model.train()