Logs

app.py

@@ -1,3 +1,4 @@
+from random import shuffle
 import streamlit as st
 
 from datasets import load_dataset
@@ -244,6 +245,8 @@ for epoch in range(num_epochs):
 
     print(f"Epoch {epoch + 1} completed with loss: {loss.item()}")
 
+    test_dataloader = DataLoader(tokenized_data['test'], batch_size=batch_size, collate_fn=data_collator, shuffle=True)
+
     # Evaluate the teacher model
     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}")