Mirrors/EOS
1
0
Fork 0
mirror of https://github.com/Akkudoktor-EOS/EOS.git synced 2025-11-04 08:46:20 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

base functions

Browse Source
This commit is contained in:
Normann
2025-01-07 00:21:43 +01:00
committed by Andreas
parent 6bcae41213
commit e0bdbdcb81
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/akkudoktoreos/prediction/elecpriceakkudoktor.py
View File

@@ -122,7 +122,7 @@ class ElecPriceAkkudoktor(ElecPriceProvider):
self.update_datetime = to_datetime(in_timezone=self.config.timezone)
return akkudoktor_data
def cap_outliers(data, sigma=2): # remove outliers
def cap_outliers(data: np.ndarray, sigma: int = 2) -> np.ndarray:
mean = data.mean()
std = data.std()
lower_bound = mean - sigma * std
@@ -131,13 +131,13 @@ class ElecPriceAkkudoktor(ElecPriceProvider):
return capped_data
def predict_ets(
history,
seasonal_periods,
):
history: np.ndarray, seasonal_periods: int, prediction_hours: int
) -> np.ndarray:
clean_history = cap_outliers(history)
model = ExponentialSmoothing(
history, seasonal="add", seasonal_periods=seasonal_periods
clean_history, seasonal="add", seasonal_periods=seasonal_periods
).fit()
return model.forecast(7 * 24)
return model.forecast(prediction_hours)
def _update_data(self, force_update: Optional[bool] = False) -> None:
"""Update forecast data in the ElecPriceDataRecord format.