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