From e67ad3779c1f8928711e8a6cd00ddfbf648e14bc Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Fri, 14 May 2021 04:12:42 +0200
Subject: [PATCH] Fix casting

---
 training.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/training.py b/training.py
index 1cf28f9..8a0f3ab 100644
--- a/training.py
+++ b/training.py
@@ -44,8 +44,8 @@ else:
                     optimizer=tf.keras.optimizers.Adam(0.001),
                     metrics=[tf.keras.metrics.RootMeanSquaredError()])
 
-    verbose = 0 if len(verbose) == 0 else verbose[0]
-    epochs = 100 if len(epochs) == 0 else epochs[0]
+    verbose = 0 if len(verbose) == 0 else int(verbose[0])
+    epochs = 100 if len(epochs) == 0 else int(epochs[0])
     model.fit(train_X, train_Y, verbose=verbose, epochs=epochs)
 
     model.save('linear_regression.h5')
\ No newline at end of file