diff --git a/docker-compose.yml b/docker-compose.yml index c8e8877..5b76f5c 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -35,9 +35,9 @@ services: DATABASE_URL: "postgres://postgres:postgres@timescaledb:5432/micrometeo?sslmode=disable" RAIN_SITE: "home" RAIN_MODEL_NAME: "rain_next_1h" - RAIN_MODEL_VERSION_BASE: "rain-logreg-v1" - RAIN_MODEL_FAMILY: "logreg" - RAIN_FEATURE_SET: "baseline" + RAIN_MODEL_VERSION_BASE: "rain-auto-v1-extended" + RAIN_MODEL_FAMILY: "auto" + RAIN_FEATURE_SET: "extended" RAIN_FORECAST_MODEL: "ecmwf" RAIN_LOOKBACK_DAYS: "30" RAIN_TRAIN_INTERVAL_HOURS: "24" diff --git a/scripts/run_rain_ml_worker.py b/scripts/run_rain_ml_worker.py index 8d48100..7ce231b 100644 --- a/scripts/run_rain_ml_worker.py +++ b/scripts/run_rain_ml_worker.py @@ -290,9 +290,9 @@ def load_config() -> WorkerConfig: database_url=database_url, site=read_env("RAIN_SITE", "home"), model_name=read_env("RAIN_MODEL_NAME", "rain_next_1h"), - model_version_base=read_env("RAIN_MODEL_VERSION_BASE", "rain-logreg-v1"), - model_family=read_env("RAIN_MODEL_FAMILY", "logreg"), - feature_set=read_env("RAIN_FEATURE_SET", "baseline"), + model_version_base=read_env("RAIN_MODEL_VERSION_BASE", "rain-auto-v1-extended"), + model_family=read_env("RAIN_MODEL_FAMILY", "auto"), + feature_set=read_env("RAIN_FEATURE_SET", "extended"), forecast_model=read_env("RAIN_FORECAST_MODEL", "ecmwf"), train_interval_hours=read_env_float("RAIN_TRAIN_INTERVAL_HOURS", 24.0), predict_interval_minutes=read_env_float("RAIN_PREDICT_INTERVAL_MINUTES", 10.0), @@ -304,7 +304,7 @@ def load_config() -> WorkerConfig: max_hyperparam_trials=read_env_int("RAIN_MAX_HYPERPARAM_TRIALS", 12), calibration_methods=read_env("RAIN_CALIBRATION_METHODS", "none,sigmoid,isotonic"), threshold_policy=read_env("RAIN_THRESHOLD_POLICY", "walk_forward"), - walk_forward_folds=read_env_int("RAIN_WALK_FORWARD_FOLDS", 0), + walk_forward_folds=read_env_int("RAIN_WALK_FORWARD_FOLDS", 4), allow_empty_data=read_env_bool("RAIN_ALLOW_EMPTY_DATA", True), dataset_path_template=read_env( "RAIN_DATASET_PATH",