From 69e6cb7bc679806619e089acc2acfdb49d240246 Mon Sep 17 00:00:00 2001
From: Sintal <61sima1bif@hft-stuttgart.de>
Date: Thu, 17 Jun 2021 14:09:56 +0000
Subject: [PATCH] Update main.py
---
main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.py b/main.py
index a0b6b2b..aeb8c73 100644
--- a/main.py
+++ b/main.py
@@ -114,7 +114,7 @@ def train_and_save_model(model, X_train, y_train, X_test, y_test):
print("Start Training...")
history = model.fit(X_train, y_train, epochs=35, validation_data=(X_test, y_test), callbacks=[tensorboard_callback])
# Saving the trained model to avoid re-training
- model.save(constants.TRAINED_MODEL)
+ #model.save(constants.TRAINED_MODEL)
return history
def model_predict(model, X_test, y_test):
--
GitLab