From 75c5a34f3c7119b03998f298d6c63a31f747c992 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sigfrid=20Stj=C3=A4rnholm?= <sigge.stjarnholm@gmail.com>
Date: Wed, 20 Jan 2021 22:22:46 +0100
Subject: [PATCH] Add modelName to on_epoch_end message

---
 tf_notification_callback/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tf_notification_callback/main.py b/tf_notification_callback/main.py
index 0bdc74f..44df105 100644
--- a/tf_notification_callback/main.py
+++ b/tf_notification_callback/main.py
@@ -62,7 +62,7 @@ def on_train_end(self, logs=None):
     def on_epoch_end(self, epoch, logs=None):
         self.logs_arr.append(logs)
         if not self.getSummary:
-            text = f'*Epoch*: {epoch}\n'
+            text = f'*{self.modelName}*:\n*Epoch*: {epoch}\n'
             for key, value in logs.items():
                 text += f'*{key}*: {value:.2f}\n'
             self.send_message(text, type='text')