update title plots

Author: JordiCorbilla

stock_prediction_plotter.py

@@ -48,6 +48,7 @@ def plot_loss(self, history):
         plt.plot(history.history['val_loss'], label='val_loss')
         plt.xlabel('Epoch')
         plt.ylabel('Loss')
+        plt.title('Loss/Validation Loss')
         plt.legend(loc='upper right')
         plt.savefig(os.path.join(self.project_folder, 'loss.png'))
         plt.pause(0.001)
@@ -59,6 +60,7 @@ def plot_mse(self, history):
         plt.plot(history.history['val_MSE'], label='val_MSE')
         plt.xlabel('Epoch')
         plt.ylabel('MSE')
+        plt.title('MSE/Validation MSE')
         plt.legend(loc='upper right')
         plt.savefig(os.path.join(self.project_folder, 'MSE.png'))
         plt.pause(0.001)
@@ -72,6 +74,7 @@ def project_plot_predictions(self, price_predicted, test_data):
         plt.xlabel('Time')
         plt.ylabel('Price [' + self.currency + ']')
         plt.legend()
+        plt.title('Prediction')
         plt.savefig(os.path.join(self.project_folder, self.short_name.strip().replace('.', '') + '_prediction.png'))
         plt.pause(0.001)
         plt.show(block=self.blocking)