From 4ac3db27ee5fba12cdd90c81fe92592be3a436b0 Mon Sep 17 00:00:00 2001
From: Daniel M <daniel.q.mueller@stud.h-da.de>
Date: Tue, 25 Jan 2022 23:57:40 +0100
Subject: [PATCH] Fix LSTM loading
---
notebooks/examples/long_short_term_memory.ipynb | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/notebooks/examples/long_short_term_memory.ipynb b/notebooks/examples/long_short_term_memory.ipynb
index 4c9ecaa..3616d25 100644
--- a/notebooks/examples/long_short_term_memory.ipynb
+++ b/notebooks/examples/long_short_term_memory.ipynb
@@ -748,11 +748,12 @@
],
"source": [
"history = None\n",
+ "epochs = 5\n",
+ "\n",
"if LOAD_MODEL:\n",
" history = pd.read_csv(BASE_PATH+'lstm_training.log', sep=',', engine='python')\n",
"else:\n",
" csv_logger = CSVLogger(BASE_PATH+'lstm_training.log', separator=',', append=False)\n",
- " epochs = 5\n",
" history=model.fit(x,y, epochs = epochs, verbose = 'auto', validation_data=(x_test, y_test), callbacks=[csv_logger]).history\n",
" model.save(BASE_PATH+'long_short_term_memory_model.h5')"
]
--
GitLab