mauriciogtec commited on
Commit
7849763
1 Parent(s): b599506

v0: add test model

Browse files
nn_full_medicare_all/config.yaml ADDED
@@ -0,0 +1,762 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ arch:
2
+ hidden_dim: 32
3
+ num_hidden_layers: 1
4
+ constraints:
5
+ baseline:
6
+ alert_lag1: negative
7
+ alerts_2wks: negative
8
+ excess_heat_3d: positive
9
+ excess_heat_7d: positive
10
+ heat_qi_above_25: positive
11
+ heat_qi_above_75: positive
12
+ hi_max_above_25: positive
13
+ hi_max_above_75: positive
14
+ effectiveness:
15
+ alert_lag1: negative
16
+ alerts_2wks: negative
17
+ excess_heat_3d: positive
18
+ excess_heat_7d: positive
19
+ heat_qi_above_25: positive
20
+ heat_qi_above_75: positive
21
+ hi_max_above_25: positive
22
+ hi_max_above_75: positive
23
+ data_dir: ../data
24
+ fips_list:
25
+ - 01049
26
+ - 01089
27
+ - '04021'
28
+ - '05145'
29
+ - '06025'
30
+ - 06079
31
+ - 06089
32
+ - 06099
33
+ - 08001
34
+ - 08005
35
+ - 08031
36
+ - '10001'
37
+ - '12083'
38
+ - '12095'
39
+ - '12103'
40
+ - '12113'
41
+ - '12119'
42
+ - '13215'
43
+ - '17115'
44
+ - '19163'
45
+ - '21029'
46
+ - '22015'
47
+ - '22033'
48
+ - '24009'
49
+ - '24013'
50
+ - '25015'
51
+ - '26099'
52
+ - '27163'
53
+ - '29095'
54
+ - '29189'
55
+ - '33015'
56
+ - '34009'
57
+ - '34025'
58
+ - '34033'
59
+ - '34035'
60
+ - '34041'
61
+ - '36013'
62
+ - '36059'
63
+ - '36063'
64
+ - '36065'
65
+ - '36071'
66
+ - '36075'
67
+ - '37101'
68
+ - '37127'
69
+ - '39061'
70
+ - '39133'
71
+ - '40027'
72
+ - '42027'
73
+ - '42033'
74
+ - '42043'
75
+ - '42049'
76
+ - '42075'
77
+ - '42129'
78
+ - '44005'
79
+ - '45015'
80
+ - '45035'
81
+ - '45041'
82
+ - '45051'
83
+ - '45091'
84
+ - '47149'
85
+ - '47165'
86
+ - '47189'
87
+ - '48041'
88
+ - '48139'
89
+ - '48347'
90
+ - '48409'
91
+ - '48469'
92
+ - '51069'
93
+ - '53007'
94
+ - '53011'
95
+ - '53021'
96
+ - '53033'
97
+ - '54003'
98
+ - '01051'
99
+ - '01055'
100
+ - 01095
101
+ - '01101'
102
+ - '01117'
103
+ - '04013'
104
+ - '04027'
105
+ - '05045'
106
+ - 05085
107
+ - '06037'
108
+ - '06041'
109
+ - '06065'
110
+ - '06071'
111
+ - 06081
112
+ - 06087
113
+ - '06101'
114
+ - '06113'
115
+ - '18053'
116
+ - '20161'
117
+ - '21093'
118
+ - '22051'
119
+ - '22057'
120
+ - '22109'
121
+ - 09001
122
+ - 09009
123
+ - 09011
124
+ - '10003'
125
+ - '10005'
126
+ - '12017'
127
+ - '12023'
128
+ - '12069'
129
+ - '12071'
130
+ - '12089'
131
+ - '13051'
132
+ - '13113'
133
+ - '13127'
134
+ - '13153'
135
+ - '13245'
136
+ - '17043'
137
+ - '17199'
138
+ - '29183'
139
+ - '30029'
140
+ - '33017'
141
+ - '34001'
142
+ - '34007'
143
+ - '34013'
144
+ - '34027'
145
+ - '34029'
146
+ - '36009'
147
+ - '36029'
148
+ - '36105'
149
+ - '37001'
150
+ - '37023'
151
+ - '37027'
152
+ - '37049'
153
+ - '37125'
154
+ - '37133'
155
+ - '37155'
156
+ - '24003'
157
+ - '24005'
158
+ - '24017'
159
+ - '24021'
160
+ - '24025'
161
+ - '24031'
162
+ - '25005'
163
+ - '25017'
164
+ - '25021'
165
+ - '25027'
166
+ - '27053'
167
+ - '27139'
168
+ - '28033'
169
+ - '28089'
170
+ - '28121'
171
+ - '29037'
172
+ - '29099'
173
+ - '47141'
174
+ - '47157'
175
+ - '47163'
176
+ - '47187'
177
+ - '48029'
178
+ - '48037'
179
+ - '48039'
180
+ - '48187'
181
+ - '48213'
182
+ - '48215'
183
+ - '48245'
184
+ - '48453'
185
+ - '51019'
186
+ - '51085'
187
+ - '51095'
188
+ - '51107'
189
+ - '51179'
190
+ - '38035'
191
+ - '39095'
192
+ - '39119'
193
+ - '40101'
194
+ - '42005'
195
+ - '42029'
196
+ - '42041'
197
+ - '42051'
198
+ - '42055'
199
+ - '42071'
200
+ - '42091'
201
+ - '42095'
202
+ - '42101'
203
+ - '42111'
204
+ - '42133'
205
+ - '44009'
206
+ - '45059'
207
+ - '45063'
208
+ - '47009'
209
+ - '47059'
210
+ - 06083
211
+ - 06085
212
+ - '06115'
213
+ - 08013
214
+ - '12001'
215
+ - '12031'
216
+ - '12033'
217
+ - '12055'
218
+ - '12073'
219
+ - '12105'
220
+ - '12117'
221
+ - '12127'
222
+ - '13021'
223
+ - '13031'
224
+ - '13077'
225
+ - '13121'
226
+ - '13135'
227
+ - '13151'
228
+ - '13313'
229
+ - '16027'
230
+ - '18157'
231
+ - '19169'
232
+ - '53041'
233
+ - '53053'
234
+ - '53057'
235
+ - '53061'
236
+ - '53067'
237
+ - '53077'
238
+ - '54011'
239
+ - '54107'
240
+ - '01003'
241
+ - '01073'
242
+ - '01121'
243
+ - '04025'
244
+ - '05007'
245
+ - '06001'
246
+ - '06007'
247
+ - '06013'
248
+ - '06023'
249
+ - '06067'
250
+ - '06073'
251
+ - '32003'
252
+ - '32031'
253
+ - '33009'
254
+ - '34003'
255
+ - '34005'
256
+ - '34021'
257
+ - '34031'
258
+ - '36051'
259
+ - '36061'
260
+ - '36067'
261
+ - '36081'
262
+ - '36091'
263
+ - '36103'
264
+ - '36119'
265
+ - '37019'
266
+ - '37119'
267
+ - '37129'
268
+ - '37147'
269
+ - '37193'
270
+ - '40125'
271
+ - '41017'
272
+ - '41033'
273
+ - '41051'
274
+ - '21067'
275
+ - '22055'
276
+ - '22073'
277
+ - '22105'
278
+ - '23005'
279
+ - '24001'
280
+ - '24033'
281
+ - '25023'
282
+ - '25025'
283
+ - '26115'
284
+ - '26163'
285
+ - '27171'
286
+ - '29019'
287
+ - '29031'
288
+ - '51165'
289
+ - '51177'
290
+ - '53015'
291
+ - '53025'
292
+ - '54039'
293
+ - '06053'
294
+ - '11001'
295
+ - '12057'
296
+ - '13217'
297
+ - '42063'
298
+ - '42097'
299
+ - '42125'
300
+ - '45007'
301
+ - '45031'
302
+ - '45045'
303
+ - '45077'
304
+ - '45085'
305
+ - '47001'
306
+ - '47037'
307
+ - '47065'
308
+ - '47093'
309
+ - '47155'
310
+ - '48061'
311
+ - '48091'
312
+ - '48135'
313
+ - '48141'
314
+ - '48157'
315
+ - '48201'
316
+ - '48303'
317
+ - '48355'
318
+ - '48471'
319
+ - '48491'
320
+ - '34023'
321
+ - '37057'
322
+ - '38017'
323
+ - '25011'
324
+ - '47147'
325
+ - '51087'
326
+ - '39043'
327
+ - '45013'
328
+ - '53073'
329
+ - 06059
330
+ - 01069
331
+ - '13073'
332
+ - '21151'
333
+ - '36053'
334
+ - '36113'
335
+ - '27137'
336
+ - '34015'
337
+ - '36047'
338
+ - '37031'
339
+ - '37085'
340
+ - '40145'
341
+ - '42003'
342
+ - '53005'
343
+ - '53027'
344
+ - 01097
345
+ - '05131'
346
+ - '06055'
347
+ - 06097
348
+ - '18035'
349
+ - 08035
350
+ - '41019'
351
+ - '01043'
352
+ - '06045'
353
+ - '37051'
354
+ - '37161'
355
+ - '22017'
356
+ - '22079'
357
+ - '45047'
358
+ - '48005'
359
+ - '48167'
360
+ - 05069
361
+ - '13185'
362
+ - '30063'
363
+ - '35045'
364
+ - '42089'
365
+ - '42017'
366
+ - '47119'
367
+ - '51059'
368
+ - '51161'
369
+ - '01115'
370
+ - '06077'
371
+ - '19013'
372
+ - '21059'
373
+ - '12097'
374
+ - '34011'
375
+ - '24037'
376
+ - '17161'
377
+ - '51041'
378
+ - '53035'
379
+ - '13013'
380
+ - '17097'
381
+ - '34019'
382
+ - '23031'
383
+ - '27145'
384
+ - '28049'
385
+ - '48203'
386
+ - '48339'
387
+ - '51061'
388
+ - '39035'
389
+ - '45003'
390
+ - '45079'
391
+ - '36045'
392
+ - '25009'
393
+ - 05119
394
+ - 06095
395
+ - '01103'
396
+ - '04003'
397
+ - '04015'
398
+ - '13015'
399
+ - '13089'
400
+ - 08123
401
+ - '13295'
402
+ - '17019'
403
+ - '13139'
404
+ - '18039'
405
+ - '18043'
406
+ - '18089'
407
+ - '18095'
408
+ - '39081'
409
+ - '20173'
410
+ - '33005'
411
+ - '39113'
412
+ - '39169'
413
+ - '35049'
414
+ - '40017'
415
+ - '28035'
416
+ - '28067'
417
+ - '36027'
418
+ - '36083'
419
+ - '29021'
420
+ - '36093'
421
+ - '36109'
422
+ - '37025'
423
+ - '37045'
424
+ - '24043'
425
+ - '37071'
426
+ - '37097'
427
+ - '23019'
428
+ - '26017'
429
+ - '37157'
430
+ - '37183'
431
+ - '26077'
432
+ - '26087'
433
+ - '26103'
434
+ - '26139'
435
+ - '26145'
436
+ - '39003'
437
+ - '20091'
438
+ - '39017'
439
+ - '26161'
440
+ - '48181'
441
+ - '55133'
442
+ - '41059'
443
+ - '48257'
444
+ - '01077'
445
+ - 01081
446
+ - '42013'
447
+ - '42069'
448
+ - '48479'
449
+ - '01125'
450
+ - '42073'
451
+ - '05031'
452
+ - '05051'
453
+ - '46099'
454
+ - '51710'
455
+ - '05143'
456
+ - '51770'
457
+ - '54033'
458
+ - '18067'
459
+ - '18081'
460
+ - '55009'
461
+ - '18163'
462
+ - '55059'
463
+ - '19049'
464
+ - '29165'
465
+ - '12081'
466
+ - '16055'
467
+ - '29187'
468
+ - '12099'
469
+ - '17001'
470
+ - '21037'
471
+ - '13059'
472
+ - '17037'
473
+ - '21117'
474
+ - '13095'
475
+ - '21145'
476
+ - '13117'
477
+ - '17089'
478
+ - '17093'
479
+ - '17119'
480
+ - '23001'
481
+ - '23011'
482
+ - '17163'
483
+ - '13247'
484
+ - '17183'
485
+ - '18005'
486
+ - '13285'
487
+ - '20045'
488
+ - '13297'
489
+ - '12005'
490
+ - '12035'
491
+ - '12053'
492
+ - '37063'
493
+ - '48027'
494
+ - '37067'
495
+ - '48099'
496
+ - '48113'
497
+ - '37135'
498
+ - '48183'
499
+ - '37159'
500
+ - '48209'
501
+ - '33011'
502
+ - '34017'
503
+ - '28075'
504
+ - '34037'
505
+ - '34039'
506
+ - '37195'
507
+ - '35001'
508
+ - '35005'
509
+ - '24015'
510
+ - '24027'
511
+ - '24045'
512
+ - '48309'
513
+ - '48329'
514
+ - '35025'
515
+ - '25013'
516
+ - '48367'
517
+ - '26049'
518
+ - '29043'
519
+ - '29047'
520
+ - '48381'
521
+ - '48397'
522
+ - '36011'
523
+ - '29077'
524
+ - '48441'
525
+ - '36019'
526
+ - '26081'
527
+ - '26091'
528
+ - '29097'
529
+ - '36085'
530
+ - '36087'
531
+ - '26111'
532
+ - '27003'
533
+ - '47125'
534
+ - '51013'
535
+ - '36117'
536
+ - '39055'
537
+ - '39063'
538
+ - '39085'
539
+ - '39089'
540
+ - '39099'
541
+ - '42045'
542
+ - '17113'
543
+ - '39109'
544
+ - '42077'
545
+ - 09003
546
+ - 09005
547
+ - '39139'
548
+ - '39141'
549
+ - '39151'
550
+ - '42085'
551
+ - 09015
552
+ - '17143'
553
+ - '44003'
554
+ - '45019'
555
+ - '45073'
556
+ - '12101'
557
+ - '18019'
558
+ - '40031'
559
+ - '18003'
560
+ - '13067'
561
+ - '17167'
562
+ - '13097'
563
+ - '13223'
564
+ - '51510'
565
+ - '51700'
566
+ - '51740'
567
+ - '40131'
568
+ - '40143'
569
+ - '16001'
570
+ - '39023'
571
+ - '39025'
572
+ - '39029'
573
+ - '39045'
574
+ - '42019'
575
+ - '42025'
576
+ - '42001'
577
+ - '37151'
578
+ - '37171'
579
+ - '37191'
580
+ - '18105'
581
+ - '38015'
582
+ - '55071'
583
+ - '55079'
584
+ - '18167'
585
+ - '18177'
586
+ - '55089'
587
+ - '55105'
588
+ - '39007'
589
+ - '39013'
590
+ - '55109'
591
+ - '55131'
592
+ - '55141'
593
+ - '39049'
594
+ - '39057'
595
+ - '19153'
596
+ - 01083
597
+ - '33013'
598
+ - '39093'
599
+ - '39101'
600
+ - '20015'
601
+ - '36001'
602
+ - '36007'
603
+ - '39103'
604
+ - 04019
605
+ - '36015'
606
+ - '39145'
607
+ - '39155'
608
+ - '51810'
609
+ - '36055'
610
+ - '39157'
611
+ - '39165'
612
+ - '39173'
613
+ - '53063'
614
+ - '05125'
615
+ - '36101'
616
+ - '40037'
617
+ - 06029
618
+ - '36111'
619
+ - '37035'
620
+ - '55027'
621
+ - '37109'
622
+ - '29051'
623
+ - '24510'
624
+ - '26005'
625
+ - '54061'
626
+ - '26025'
627
+ - '26045'
628
+ - '26065'
629
+ - '26075'
630
+ - '12115'
631
+ - '26093'
632
+ - '55025'
633
+ - '55039'
634
+ - '55055'
635
+ - '13057'
636
+ - '13115'
637
+ - '55117'
638
+ - '26147'
639
+ - '55127'
640
+ - '26159'
641
+ - '55139'
642
+ - '20103'
643
+ - '27037'
644
+ - '27109'
645
+ - '21015'
646
+ - '27141'
647
+ - '51199'
648
+ - '21047'
649
+ - '51680'
650
+ - '51760'
651
+ - '21227'
652
+ - '48251'
653
+ - '47179'
654
+ - '48021'
655
+ - '25003'
656
+ - '42007'
657
+ - '48085'
658
+ - '48423'
659
+ - '48451'
660
+ - '26073'
661
+ - '17111'
662
+ - '48121'
663
+ - '51650'
664
+ - '18059'
665
+ - '48231'
666
+ - '19113'
667
+ - '37081'
668
+ - '21111'
669
+ - '48291'
670
+ - '42039'
671
+ - '42021'
672
+ - '42037'
673
+ - '48375'
674
+ - '26155'
675
+ - '42079'
676
+ - '48485'
677
+ - '42107'
678
+ - '44007'
679
+ - '31109'
680
+ - '55073'
681
+ - '31055'
682
+ - '31153'
683
+ - '36005'
684
+ - '26037'
685
+ - '37179'
686
+ - '01015'
687
+ - 09007
688
+ - '12091'
689
+ - '19103'
690
+ - '22097'
691
+ - '18063'
692
+ - '18127'
693
+ - '55101'
694
+ - '12109'
695
+ - '13045'
696
+ - '17091'
697
+ - '18097'
698
+ - '17201'
699
+ - '36089'
700
+ - '27123'
701
+ - '28081'
702
+ - '29510'
703
+ - '40109'
704
+ - '41029'
705
+ - '36079'
706
+ - '26055'
707
+ - '39041'
708
+ - '40119'
709
+ - '35013'
710
+ - '35061'
711
+ - '26125'
712
+ - '47113'
713
+ - '17099'
714
+ - '48439'
715
+ - '55087'
716
+ - '39153'
717
+ - '46103'
718
+ - '51800'
719
+ - '55035'
720
+ - '06075'
721
+ - 09013
722
+ - '13063'
723
+ - '47011'
724
+ - '01127'
725
+ - '18057'
726
+ - '19193'
727
+ - '18109'
728
+ - '55063'
729
+ - '19061'
730
+ - '29071'
731
+ - '20177'
732
+ - '12015'
733
+ - '20209'
734
+ - '50007'
735
+ - '17179'
736
+ - '25001'
737
+ - '26121'
738
+ - '55097'
739
+ - '27019'
740
+ - '42011'
741
+ - '51550'
742
+ - '42081'
743
+ - '36069'
744
+ - '19155'
745
+ hospitalizations:
746
+ _target_: hospitalizations.load_hosps
747
+ data_path: ../data/hospitalizations/medicare.parquet
748
+ name: medicare
749
+ name: nn_full_medicare_all
750
+ num_samples: 100
751
+ training:
752
+ accelerator: cpu
753
+ batch_size: null
754
+ epochs: 100
755
+ gradient_clip_val: 1.0
756
+ jit: true
757
+ lr: 0.003
758
+ lr_gamma: 0.1
759
+ lr_step: 30
760
+ max_steps: -1
761
+ num_particles: 3
762
+ num_workers: 0
nn_full_medicare_all/posterior_samples.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:443e8b876e0b29bd680f66eb38cc3106feb0d00f762c70a1aca2dd8b1fd24318
3
+ size 16133432