loss

Code/loss_visualization.py

@@ -2,16 +2,29 @@
 
 import matplotlib.pyplot as plt
 import re
+import numpy as np
 
 with open("output_loss_epoch_data.txt", "r") as f:
     loss_file = f.read()
 
-pattern = r"\((\d+\.\d+)"
+loss_pattern = r"\((\d+\.\d+)"
 
-matches = re.findall(pattern, loss_file)
-print(matches)
+epochs_pattern = r"Epoche: *(\d+)"
 
+loss_values = re.findall(loss_pattern, loss_file)
+#np.array(loss_values)
+#print(loss_values)
 
-plt.plot()
+epochs = re.findall(epochs_pattern, loss_file)
+#np.array(epochs)
+#print(epochs)
+
+    
+plt.ylabel("Loss")
+plt.xlabel("Epochs")
+plt.title("Loss of Baselines")
+#plt.legend()
+
+plt.plot(epochs, loss_values[:25])
 plt.show()