alexander/urban-meal-delivery
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity Actions

Adjust OrderHistory.choose_tactical_model() heuristic

Browse source 
- use the `HorizontalSMAModel` for low demand
- use the `TrivialModel` for no demand
This commit is contained in:
Alexander Hess 2021-02-02 15:20:02 +01:00
parent 3f5b4a50bb
commit 23391c2fa4
Signed by: alexander
GPG key ID: 344EA5AB10D868E0
2 changed files with 4 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/urban_meal_delivery/forecasts/timify.py
View file

@ -545,12 +545,10 @@ class OrderHistory:
elif add >= 10: # = "medium demand"
return models.HorizontalETSModel(order_history=self)
elif add >= 2.5: # = "low demand"
# TODO: create HorizontalSMAModel
return models.HorizontalETSModel(order_history=self)
return models.HorizontalSMAModel(order_history=self)
# = "no demand"
# TODO: create HorizontalTrivialModel
return models.HorizontalETSModel(order_history=self)
return models.TrivialModel(order_history=self)
raise RuntimeError(
'no rule for the given average daily demand and training horizon',

6
tests/forecasts/timify/test_avg_daily_demand.py
View file

@ -112,8 +112,7 @@ class TestChooseTacticalModel:
train_horizon=test_config.LONG_TRAIN_HORIZON,
)
# TODO: this should be the future `HorizontalSMAModel`.
assert isinstance(result, models.HorizontalETSModel)
assert isinstance(result, models.HorizontalSMAModel)
def test_best_model_with_no_demand(
self, order_history, good_pixel_id, predict_at,
@ -127,8 +126,7 @@ class TestChooseTacticalModel:
train_horizon=test_config.LONG_TRAIN_HORIZON,
)
# TODO: this should be the future `HorizontalTrivialModel`.
assert isinstance(result, models.HorizontalETSModel)
assert isinstance(result, models.TrivialModel)
def test_best_model_for_unknown_train_horizon(
self, order_history, good_pixel_id, predict_at, # noqa:RST215