diff --git a/src/akkudoktoreos/prediction/price_forecast.py b/src/akkudoktoreos/prediction/price_forecast.py index 059f833..7b4fd58 100644 --- a/src/akkudoktoreos/prediction/price_forecast.py +++ b/src/akkudoktoreos/prediction/price_forecast.py @@ -164,6 +164,7 @@ class HourlyElectricityPriceForecast: axis=0, weights=np.array([1, 2, 4, 8, 16, 32, 64]) / np.sum(np.array([1, 2, 4, 8, 16, 32, 64])), ) + final_weights = np.linspace(1, 0, price_matrix.shape[1]) # Weight last known price linear falling @@ -175,6 +176,7 @@ class HourlyElectricityPriceForecast: return np.array(average_prices_with_final_weight) + def get_price_for_daterange( self, start_date_str: str, end_date_str: str, repeat: bool = False ) -> np.ndarray: