--- a
+++ b/generate_plot.py
@@ -0,0 +1,24 @@
+import pandas as pd
+import numpy as np
+import matplotlib.pyplot as plt
+
+value_list = np.zeros((1000, 5))
+
+for i in range(5):
+    fields = ['Step', 'Value']
+    df = pd.read_csv('./checkpoints/run-train-tag-batch_loss_' + str(i + 1) + '.csv', usecols=fields)
+
+    steps = df.Step.to_numpy()
+    value = df.Value.to_numpy()
+
+    value_list[:, i] = value
+
+value_mean = np.mean(value_list, axis=1)
+value_std = np.std(value_list, axis=1)
+
+plt.plot(steps, value_mean)
+plt.xlabel('Train Step')
+plt.ylabel('Tversky Loss')
+plt.fill_between(steps, value_mean - value_std, value_mean + value_std, alpha=.5)
+plt.grid()
+plt.show()