From d21f632de3582de26a3dcbc5e9720fbe7c36b855 Mon Sep 17 00:00:00 2001
From: Ludovic Moncla <ludovic.moncla@insa-lyon.fr>
Date: Thu, 10 Dec 2020 17:59:23 +0100
Subject: [PATCH] Update train_geocoder.py
---
train_geocoder.py | 5 +----
1 file changed, 1 insertion(+), 4 deletions(-)
diff --git a/train_geocoder.py b/train_geocoder.py
index 4b7bfc3..c2c15e8 100644
--- a/train_geocoder.py
+++ b/train_geocoder.py
@@ -203,7 +203,7 @@ print(model.summary())
checkpoint = ModelCheckpoint(MODEL_OUTPUT_FN + ".part", monitor='loss', verbose=1,
save_best_only=True, mode='auto', period=1)
-epoch_timer = EpochTimer("outputs/"+PREFIX_OUTPUT_FN+"_epoch_timer_output.csv")
+epoch_timer = EpochTimer(HISTORY_FN)
@@ -213,9 +213,6 @@ history = model.fit(training_generator,verbose=True,
-hist_df = pd.DataFrame(history.history)
-hist_df.to_csv(HISTORY_FN)
-
model.save(MODEL_OUTPUT_FN)
#Â Erase Model Checkpoint file
--
GitLab