loss

Code/loss_visualization.py

@@ -3,7 +3,7 @@
 import matplotlib.pyplot as plt
 import re
 
-with open("output_baselines_loss.txt", "r") as f:
+with open("output_loss_epoch_data.txt", "r") as f:
     loss_file = f.read()
 
 pattern = r"\((\d+\.\d+)"
@@ -12,6 +12,6 @@ matches = re.findall(pattern, loss_file)
 print(matches)
     
 
-plt.plot(matches)
+plt.plot()
 plt.show()