Files
Enelix-EMS/services/netplan-v4/acceptance/forecast-src/methods/var_3.py
T

14 lines
416 B
Python

from methods.battery_optimizer import optimize_grid_setpoint, train_artifact
from methods.common import save_model
FORECAST_ID = 3
def train(data_obj):
aid = data_obj["config"]["anlagen_id"]
return {"trained": True, "path": save_model(aid, FORECAST_ID, train_artifact("var_1_2"))}
def predict(data_obj, pv_dict, load_dict):
return optimize_grid_setpoint(data_obj, pv_dict, load_dict, forecast_id=3)