From 789b65623a0d4ce43461a9a7c2ac0e4a74e0703e Mon Sep 17 00:00:00 2001
From: "v.gramlich1" <v.gramlichfz-juelich.de>
Date: Tue, 29 Jun 2021 16:12:38 +0200
Subject: [PATCH] Set-up to run with and without oversampling on hdfml, full
 stationlist, 150epochs

---
 run_with_oversampling.py    | 3 ++-
 run_without_oversampling.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/run_with_oversampling.py b/run_with_oversampling.py
index b21e5e6f..023a8a6f 100644
--- a/run_with_oversampling.py
+++ b/run_with_oversampling.py
@@ -26,8 +26,9 @@ def load_stations(external_station_list):
 
 def main(parser_args):
     plots = remove_items(DEFAULT_PLOT_LIST, "PlotConditionalQuantiles")
-    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json')[:75],
+    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json'),
         #stations=["DEBW013", "DEBW087", "DEBW107", "DEBW076"],
+        epochs=150,
         train_model=True, create_new_model=True, network="UBA",
         model=IntelliO3_ts_architecture, oversampling_method="bin_oversampling",
         evaluate_bootstraps=False,  # plot_list=["PlotCompetitiveSkillScore"],
diff --git a/run_without_oversampling.py b/run_without_oversampling.py
index 5b51ffa0..bda899ca 100644
--- a/run_without_oversampling.py
+++ b/run_without_oversampling.py
@@ -26,8 +26,9 @@ def load_stations(external_station_list = None):
 
 def main(parser_args):
     plots = remove_items(DEFAULT_PLOT_LIST, "PlotConditionalQuantiles")
-    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json')[:75],
+    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_station.json'),
         #stations=["DEBW013", "DEBW087", "DEBW107", "DEBW076"],
+        epochs=150,
         train_model=True, create_new_model=True, network="UBA",
         model=IntelliO3_ts_architecture,
         evaluate_bootstraps=False,  # plot_list=["PlotCompetitiveSkillScore"],
-- 
GitLab