deal with training horizons larger than 15 days

Signed-off-by: Martijn Cazemier <[email protected]>
OpenSTEF · Dec 7, 2023 · 69a1098 · 69a1098
1 parent c1cc311
commit 69a1098
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openstef/feature_engineering/lag_features.py b/openstef/feature_engineering/lag_features.py
@@ -108,7 +108,7 @@ def generate_trivial_lag_features(horizon: float) -> tuple[list, list]:
         - List of days lags that were used as features during training.
 
     """
-    mindays = int(np.ceil(horizon / 24))
+    mindays = min(int(np.ceil(horizon / 24)), 15)
     lag_time_days_list = list(np.linspace(mindays, 14, 15 - mindays))
 
     # Make list of trivial lag times