Initial commit

ui/plotters/gradients_plotter.py

@@ -0,0 +1,7 @@
+from abc import ABC
+
+from matplotlib.figure import Figure
+
+from neural_net.epoch import Epoch
+from neural_net.neural_net import NeuralNet
+from ui.plotters.plotter import Plotter

ui/plotters/loss_plotter.py

@@ -0,0 +1 @@
+

ui/plotters/plotter.py

@@ -0,0 +1,30 @@
+from abc import abstractmethod
+
+from matplotlib.figure import Figure
+
+from neural_net.epoch import Epoch
+
+
+class Plotter:
+    def __init__(self, figure: Figure):
+        self.figure = figure
+
+    def initialize_plots(self):
+        self.figure.show()
+
+    @abstractmethod
+    def update_plot(self, data):
+        self.reset_plot()
+
+        self.plot(data)
+
+        self.figure.canvas.draw()
+        self.figure.canvas.flush_events()
+
+    @abstractmethod
+    def reset_plot(self):
+        pass
+
+    @abstractmethod
+    def plot(self, current_epoch: Epoch):
+        pass