mauriciogtec's picture
v0: add test model
7849763
arch:
hidden_dim: 32
num_hidden_layers: 1
constraints:
baseline:
alert_lag1: negative
alerts_2wks: negative
excess_heat_3d: positive
excess_heat_7d: positive
heat_qi_above_25: positive
heat_qi_above_75: positive
hi_max_above_25: positive
hi_max_above_75: positive
effectiveness:
alert_lag1: negative
alerts_2wks: negative
excess_heat_3d: positive
excess_heat_7d: positive
heat_qi_above_25: positive
heat_qi_above_75: positive
hi_max_above_25: positive
hi_max_above_75: positive
data_dir: ../data
fips_list:
- 01049
- 01089
- '04021'
- '05145'
- '06025'
- 06079
- 06089
- 06099
- 08001
- 08005
- 08031
- '10001'
- '12083'
- '12095'
- '12103'
- '12113'
- '12119'
- '13215'
- '17115'
- '19163'
- '21029'
- '22015'
- '22033'
- '24009'
- '24013'
- '25015'
- '26099'
- '27163'
- '29095'
- '29189'
- '33015'
- '34009'
- '34025'
- '34033'
- '34035'
- '34041'
- '36013'
- '36059'
- '36063'
- '36065'
- '36071'
- '36075'
- '37101'
- '37127'
- '39061'
- '39133'
- '40027'
- '42027'
- '42033'
- '42043'
- '42049'
- '42075'
- '42129'
- '44005'
- '45015'
- '45035'
- '45041'
- '45051'
- '45091'
- '47149'
- '47165'
- '47189'
- '48041'
- '48139'
- '48347'
- '48409'
- '48469'
- '51069'
- '53007'
- '53011'
- '53021'
- '53033'
- '54003'
- '01051'
- '01055'
- 01095
- '01101'
- '01117'
- '04013'
- '04027'
- '05045'
- 05085
- '06037'
- '06041'
- '06065'
- '06071'
- 06081
- 06087
- '06101'
- '06113'
- '18053'
- '20161'
- '21093'
- '22051'
- '22057'
- '22109'
- 09001
- 09009
- 09011
- '10003'
- '10005'
- '12017'
- '12023'
- '12069'
- '12071'
- '12089'
- '13051'
- '13113'
- '13127'
- '13153'
- '13245'
- '17043'
- '17199'
- '29183'
- '30029'
- '33017'
- '34001'
- '34007'
- '34013'
- '34027'
- '34029'
- '36009'
- '36029'
- '36105'
- '37001'
- '37023'
- '37027'
- '37049'
- '37125'
- '37133'
- '37155'
- '24003'
- '24005'
- '24017'
- '24021'
- '24025'
- '24031'
- '25005'
- '25017'
- '25021'
- '25027'
- '27053'
- '27139'
- '28033'
- '28089'
- '28121'
- '29037'
- '29099'
- '47141'
- '47157'
- '47163'
- '47187'
- '48029'
- '48037'
- '48039'
- '48187'
- '48213'
- '48215'
- '48245'
- '48453'
- '51019'
- '51085'
- '51095'
- '51107'
- '51179'
- '38035'
- '39095'
- '39119'
- '40101'
- '42005'
- '42029'
- '42041'
- '42051'
- '42055'
- '42071'
- '42091'
- '42095'
- '42101'
- '42111'
- '42133'
- '44009'
- '45059'
- '45063'
- '47009'
- '47059'
- 06083
- 06085
- '06115'
- 08013
- '12001'
- '12031'
- '12033'
- '12055'
- '12073'
- '12105'
- '12117'
- '12127'
- '13021'
- '13031'
- '13077'
- '13121'
- '13135'
- '13151'
- '13313'
- '16027'
- '18157'
- '19169'
- '53041'
- '53053'
- '53057'
- '53061'
- '53067'
- '53077'
- '54011'
- '54107'
- '01003'
- '01073'
- '01121'
- '04025'
- '05007'
- '06001'
- '06007'
- '06013'
- '06023'
- '06067'
- '06073'
- '32003'
- '32031'
- '33009'
- '34003'
- '34005'
- '34021'
- '34031'
- '36051'
- '36061'
- '36067'
- '36081'
- '36091'
- '36103'
- '36119'
- '37019'
- '37119'
- '37129'
- '37147'
- '37193'
- '40125'
- '41017'
- '41033'
- '41051'
- '21067'
- '22055'
- '22073'
- '22105'
- '23005'
- '24001'
- '24033'
- '25023'
- '25025'
- '26115'
- '26163'
- '27171'
- '29019'
- '29031'
- '51165'
- '51177'
- '53015'
- '53025'
- '54039'
- '06053'
- '11001'
- '12057'
- '13217'
- '42063'
- '42097'
- '42125'
- '45007'
- '45031'
- '45045'
- '45077'
- '45085'
- '47001'
- '47037'
- '47065'
- '47093'
- '47155'
- '48061'
- '48091'
- '48135'
- '48141'
- '48157'
- '48201'
- '48303'
- '48355'
- '48471'
- '48491'
- '34023'
- '37057'
- '38017'
- '25011'
- '47147'
- '51087'
- '39043'
- '45013'
- '53073'
- 06059
- 01069
- '13073'
- '21151'
- '36053'
- '36113'
- '27137'
- '34015'
- '36047'
- '37031'
- '37085'
- '40145'
- '42003'
- '53005'
- '53027'
- 01097
- '05131'
- '06055'
- 06097
- '18035'
- 08035
- '41019'
- '01043'
- '06045'
- '37051'
- '37161'
- '22017'
- '22079'
- '45047'
- '48005'
- '48167'
- 05069
- '13185'
- '30063'
- '35045'
- '42089'
- '42017'
- '47119'
- '51059'
- '51161'
- '01115'
- '06077'
- '19013'
- '21059'
- '12097'
- '34011'
- '24037'
- '17161'
- '51041'
- '53035'
- '13013'
- '17097'
- '34019'
- '23031'
- '27145'
- '28049'
- '48203'
- '48339'
- '51061'
- '39035'
- '45003'
- '45079'
- '36045'
- '25009'
- 05119
- 06095
- '01103'
- '04003'
- '04015'
- '13015'
- '13089'
- 08123
- '13295'
- '17019'
- '13139'
- '18039'
- '18043'
- '18089'
- '18095'
- '39081'
- '20173'
- '33005'
- '39113'
- '39169'
- '35049'
- '40017'
- '28035'
- '28067'
- '36027'
- '36083'
- '29021'
- '36093'
- '36109'
- '37025'
- '37045'
- '24043'
- '37071'
- '37097'
- '23019'
- '26017'
- '37157'
- '37183'
- '26077'
- '26087'
- '26103'
- '26139'
- '26145'
- '39003'
- '20091'
- '39017'
- '26161'
- '48181'
- '55133'
- '41059'
- '48257'
- '01077'
- 01081
- '42013'
- '42069'
- '48479'
- '01125'
- '42073'
- '05031'
- '05051'
- '46099'
- '51710'
- '05143'
- '51770'
- '54033'
- '18067'
- '18081'
- '55009'
- '18163'
- '55059'
- '19049'
- '29165'
- '12081'
- '16055'
- '29187'
- '12099'
- '17001'
- '21037'
- '13059'
- '17037'
- '21117'
- '13095'
- '21145'
- '13117'
- '17089'
- '17093'
- '17119'
- '23001'
- '23011'
- '17163'
- '13247'
- '17183'
- '18005'
- '13285'
- '20045'
- '13297'
- '12005'
- '12035'
- '12053'
- '37063'
- '48027'
- '37067'
- '48099'
- '48113'
- '37135'
- '48183'
- '37159'
- '48209'
- '33011'
- '34017'
- '28075'
- '34037'
- '34039'
- '37195'
- '35001'
- '35005'
- '24015'
- '24027'
- '24045'
- '48309'
- '48329'
- '35025'
- '25013'
- '48367'
- '26049'
- '29043'
- '29047'
- '48381'
- '48397'
- '36011'
- '29077'
- '48441'
- '36019'
- '26081'
- '26091'
- '29097'
- '36085'
- '36087'
- '26111'
- '27003'
- '47125'
- '51013'
- '36117'
- '39055'
- '39063'
- '39085'
- '39089'
- '39099'
- '42045'
- '17113'
- '39109'
- '42077'
- 09003
- 09005
- '39139'
- '39141'
- '39151'
- '42085'
- 09015
- '17143'
- '44003'
- '45019'
- '45073'
- '12101'
- '18019'
- '40031'
- '18003'
- '13067'
- '17167'
- '13097'
- '13223'
- '51510'
- '51700'
- '51740'
- '40131'
- '40143'
- '16001'
- '39023'
- '39025'
- '39029'
- '39045'
- '42019'
- '42025'
- '42001'
- '37151'
- '37171'
- '37191'
- '18105'
- '38015'
- '55071'
- '55079'
- '18167'
- '18177'
- '55089'
- '55105'
- '39007'
- '39013'
- '55109'
- '55131'
- '55141'
- '39049'
- '39057'
- '19153'
- 01083
- '33013'
- '39093'
- '39101'
- '20015'
- '36001'
- '36007'
- '39103'
- 04019
- '36015'
- '39145'
- '39155'
- '51810'
- '36055'
- '39157'
- '39165'
- '39173'
- '53063'
- '05125'
- '36101'
- '40037'
- 06029
- '36111'
- '37035'
- '55027'
- '37109'
- '29051'
- '24510'
- '26005'
- '54061'
- '26025'
- '26045'
- '26065'
- '26075'
- '12115'
- '26093'
- '55025'
- '55039'
- '55055'
- '13057'
- '13115'
- '55117'
- '26147'
- '55127'
- '26159'
- '55139'
- '20103'
- '27037'
- '27109'
- '21015'
- '27141'
- '51199'
- '21047'
- '51680'
- '51760'
- '21227'
- '48251'
- '47179'
- '48021'
- '25003'
- '42007'
- '48085'
- '48423'
- '48451'
- '26073'
- '17111'
- '48121'
- '51650'
- '18059'
- '48231'
- '19113'
- '37081'
- '21111'
- '48291'
- '42039'
- '42021'
- '42037'
- '48375'
- '26155'
- '42079'
- '48485'
- '42107'
- '44007'
- '31109'
- '55073'
- '31055'
- '31153'
- '36005'
- '26037'
- '37179'
- '01015'
- 09007
- '12091'
- '19103'
- '22097'
- '18063'
- '18127'
- '55101'
- '12109'
- '13045'
- '17091'
- '18097'
- '17201'
- '36089'
- '27123'
- '28081'
- '29510'
- '40109'
- '41029'
- '36079'
- '26055'
- '39041'
- '40119'
- '35013'
- '35061'
- '26125'
- '47113'
- '17099'
- '48439'
- '55087'
- '39153'
- '46103'
- '51800'
- '55035'
- '06075'
- 09013
- '13063'
- '47011'
- '01127'
- '18057'
- '19193'
- '18109'
- '55063'
- '19061'
- '29071'
- '20177'
- '12015'
- '20209'
- '50007'
- '17179'
- '25001'
- '26121'
- '55097'
- '27019'
- '42011'
- '51550'
- '42081'
- '36069'
- '19155'
hospitalizations:
_target_: hospitalizations.load_hosps
data_path: ../data/hospitalizations/medicare.parquet
name: medicare
name: nn_full_medicare_all
num_samples: 100
training:
accelerator: cpu
batch_size: null
epochs: 100
gradient_clip_val: 1.0
jit: true
lr: 0.003
lr_gamma: 0.1
lr_step: 30
max_steps: -1
num_particles: 3
num_workers: 0