Skip to content

Instantly share code, notes, and snippets.

@hassanj47

hassanj47/ts_survival_config.py

Last active June 6, 2022 20:36
Show Gist options
  • Save hassanj47/5b10e02221608798fd2898136f0f740f to your computer and use it in GitHub Desktop.
Save hassanj47/5b10e02221608798fd2898136f0f740f to your computer and use it in GitHub Desktop.
Download ZIP
Configuration for time series survival model
Raw
ts_survival_config.py
# CONFIG
TRAIN_END = 'd_1913'
DEPTS = ['HOBBIES_1', 'HOBBIES_2', 'FOODS_1','FOODS_2', 'FOODS_3']
STORES = ['CA_1' ,'CA_2', 'CA_3', 'CA_4', 'TX_1', 'TX_2', 'TX_3', 'WI_1',
'WI_2', 'WI_3']
THRESH = 10 # MAPE threshold
TEST_DAYS = 365
FORECAST_DAYS = 365
WIN_SIZE_WEEKS = 4 # For expanding MAPE
STUDY_START_WEEKS = WIN_SIZE_WEEKS
STUDY_END_WEEKS = 26
SURVIVAL_PERCENTILE = 0.5
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment