DTanzillo commited on
Commit
d909dd9
·
verified ·
1 Parent(s): e1e5bcf

Upload results/deep_model_results.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. results/deep_model_results.json +669 -489
results/deep_model_results.json CHANGED
@@ -1,490 +1,670 @@
1
- {
2
- "model": "PI-TFT (Physics-Informed Temporal Fusion Transformer)",
3
- "best_epoch": 41,
4
- "training_time_minutes": 14.84879409869512,
5
- "optimal_threshold": 0.36438190937042236,
6
- "temperature": 0.6953535079956055,
7
- "test": {
8
- "loss": 0.023227682939785367,
9
- "auc_pr": 0.6077793231111472,
10
- "auc_roc": 0.9728513299577397,
11
- "f1_at_50": 0.3111111111111111,
12
- "n_positive": 73,
13
- "n_total": 2167,
14
- "pos_rate": 0.03368712589144707,
15
- "f1": 0.5798816518973426,
16
- "optimal_threshold": 0.36438190937042236,
17
- "threshold": 0.36438190937042236,
18
- "recall_at_prec_30": 0.863013698630137,
19
- "recall_at_prec_50": 0.6712328767123288,
20
- "recall_at_prec_70": 0.3424657534246575,
21
- "mae_log": 0.2623131573200226,
22
- "rmse_log": 0.3380082758004626,
23
- "mae_km": 3949.743408203125,
24
- "median_abs_error_km": 2327.5947265625
25
- },
26
- "test_calibrated": {
27
- "auc_pr": 0.6077793231111472,
28
- "auc_roc": 0.9728513299577397,
29
- "f1_at_50": 0.3111111111111111,
30
- "n_positive": 73,
31
- "n_total": 2167,
32
- "pos_rate": 0.03368712589144707,
33
- "f1": 0.5798816518973426,
34
- "optimal_threshold": 0.30999091267585754,
35
- "threshold": 0.30999091267585754,
36
- "recall_at_prec_30": 0.863013698630137,
37
- "recall_at_prec_50": 0.6712328767123288,
38
- "recall_at_prec_70": 0.3424657534246575
39
- },
40
- "history": [
41
- {
42
- "epoch": 1,
43
- "train_loss": 6.043865992571856,
44
- "val_loss": 4.417774633927778,
45
- "val_auc_pr": 0.005531670674240891,
46
- "val_f1": 0.013234077618741713,
47
- "val_mae_log": 6.491847515106201
48
- },
49
- {
50
- "epoch": 2,
51
- "train_loss": 3.386568644240096,
52
- "val_loss": 2.4896007884632456,
53
- "val_auc_pr": 0.011403126957381278,
54
- "val_f1": 0.05263157706371198,
55
- "val_mae_log": 4.803258895874023
56
- },
57
- {
58
- "epoch": 3,
59
- "train_loss": 1.8177962857323724,
60
- "val_loss": 1.0469709092920476,
61
- "val_auc_pr": 0.020608162739287648,
62
- "val_f1": 0.09999999520000023,
63
- "val_mae_log": 2.987166404724121
64
- },
65
- {
66
- "epoch": 4,
67
- "train_loss": 0.5769605080823641,
68
- "val_loss": 0.17964649200439453,
69
- "val_auc_pr": 0.1544556359138917,
70
- "val_f1": 0.2499999955555556,
71
- "val_mae_log": 1.1048060655593872
72
- },
73
- {
74
- "epoch": 5,
75
- "train_loss": 0.12293093365591926,
76
- "val_loss": 0.03570454415272583,
77
- "val_auc_pr": 0.1614990598336613,
78
- "val_f1": 0.2727272680991736,
79
- "val_mae_log": 0.3387407958507538
80
- },
81
- {
82
- "epoch": 6,
83
- "train_loss": 0.07903797459360716,
84
- "val_loss": 0.022081364708190613,
85
- "val_auc_pr": 0.2123489041200941,
86
- "val_f1": 0.3999999950222222,
87
- "val_mae_log": 0.25661933422088623
88
- },
89
- {
90
- "epoch": 7,
91
- "train_loss": 0.06960436115796502,
92
- "val_loss": 0.0379494982348247,
93
- "val_auc_pr": 0.2575112899590874,
94
- "val_f1": 0.37499999500000003,
95
- "val_mae_log": 0.4238395392894745
96
- },
97
- {
98
- "epoch": 8,
99
- "train_loss": 0.06403992270698418,
100
- "val_loss": 0.023355446925217457,
101
- "val_auc_pr": 0.19355434476368538,
102
- "val_f1": 0.3448275822116528,
103
- "val_mae_log": 0.2827949821949005
104
- },
105
- {
106
- "epoch": 9,
107
- "train_loss": 0.06516141986122002,
108
- "val_loss": 0.02040538191795349,
109
- "val_auc_pr": 0.198119511085929,
110
- "val_f1": 0.3448275822116528,
111
- "val_mae_log": 0.27033379673957825
112
- },
113
- {
114
- "epoch": 10,
115
- "train_loss": 0.06380475181582812,
116
- "val_loss": 0.021484596146778626,
117
- "val_auc_pr": 0.15521974237838998,
118
- "val_f1": 0.28571428099773244,
119
- "val_mae_log": 0.2632426917552948
120
- },
121
- {
122
- "epoch": 11,
123
- "train_loss": 0.05905778277161959,
124
- "val_loss": 0.021725460632958195,
125
- "val_auc_pr": 0.23805690836940838,
126
- "val_f1": 0.4444444395061729,
127
- "val_mae_log": 0.33999449014663696
128
- },
129
- {
130
- "epoch": 12,
131
- "train_loss": 0.05756706851156982,
132
- "val_loss": 0.0313187607310035,
133
- "val_auc_pr": 0.14051089510648335,
134
- "val_f1": 0.23999999564800004,
135
- "val_mae_log": 0.34385421872138977
136
- },
137
- {
138
- "epoch": 13,
139
- "train_loss": 0.05848575969402855,
140
- "val_loss": 0.014629591425711458,
141
- "val_auc_pr": 0.11825738348638665,
142
- "val_f1": 0.26666666275555556,
143
- "val_mae_log": 0.19728681445121765
144
- },
145
- {
146
- "epoch": 14,
147
- "train_loss": 0.05825336039871783,
148
- "val_loss": 0.016250551847571678,
149
- "val_auc_pr": 0.1715640264305125,
150
- "val_f1": 0.37037036620027436,
151
- "val_mae_log": 0.27703016996383667
152
- },
153
- {
154
- "epoch": 15,
155
- "train_loss": 0.053515979939618624,
156
- "val_loss": 0.015867904294282198,
157
- "val_auc_pr": 0.19873776585733108,
158
- "val_f1": 0.3478260824196598,
159
- "val_mae_log": 0.2324916124343872
160
- },
161
- {
162
- "epoch": 16,
163
- "train_loss": 0.05182460594821621,
164
- "val_loss": 0.012541103261438284,
165
- "val_auc_pr": 0.1811582643541063,
166
- "val_f1": 0.3414634114931589,
167
- "val_mae_log": 0.20959354937076569
168
- },
169
- {
170
- "epoch": 17,
171
- "train_loss": 0.05259045489333771,
172
- "val_loss": 0.01332802198488604,
173
- "val_auc_pr": 0.2790156915156915,
174
- "val_f1": 0.319999997312,
175
- "val_mae_log": 0.19698341190814972
176
- },
177
- {
178
- "epoch": 18,
179
- "train_loss": 0.05047502589185496,
180
- "val_loss": 0.016544518992304802,
181
- "val_auc_pr": 0.1815782353611301,
182
- "val_f1": 0.2962962921262003,
183
- "val_mae_log": 0.23860619962215424
184
- },
185
- {
186
- "epoch": 19,
187
- "train_loss": 0.04817935439783173,
188
- "val_loss": 0.02867866442962126,
189
- "val_auc_pr": 0.2761163985475398,
190
- "val_f1": 0.3999999960888889,
191
- "val_mae_log": 0.37167781591415405
192
- },
193
- {
194
- "epoch": 20,
195
- "train_loss": 0.05029433012210034,
196
- "val_loss": 0.010146147444505583,
197
- "val_auc_pr": 0.2814331501831502,
198
- "val_f1": 0.4285714244897959,
199
- "val_mae_log": 0.15699289739131927
200
- },
201
- {
202
- "epoch": 21,
203
- "train_loss": 0.05035827603895922,
204
- "val_loss": 0.010205186734145338,
205
- "val_auc_pr": 0.22964399198414542,
206
- "val_f1": 0.39024389929803693,
207
- "val_mae_log": 0.1536967009305954
208
- },
209
- {
210
- "epoch": 22,
211
- "train_loss": 0.05115810244067295,
212
- "val_loss": 0.014916623519225554,
213
- "val_auc_pr": 0.19813229479202776,
214
- "val_f1": 0.3555555526320988,
215
- "val_mae_log": 0.27219313383102417
216
- },
217
- {
218
- "epoch": 23,
219
- "train_loss": 0.047980119630291654,
220
- "val_loss": 0.011064541526138783,
221
- "val_auc_pr": 0.14539742773168318,
222
- "val_f1": 0.30769230508875745,
223
- "val_mae_log": 0.16786827147006989
224
- },
225
- {
226
- "epoch": 24,
227
- "train_loss": 0.04663550586917916,
228
- "val_loss": 0.009120341602035544,
229
- "val_auc_pr": 0.33940970983752866,
230
- "val_f1": 0.38095237786848074,
231
- "val_mae_log": 0.1796921193599701
232
- },
233
- {
234
- "epoch": 25,
235
- "train_loss": 0.04705044173106954,
236
- "val_loss": 0.009002900174395605,
237
- "val_auc_pr": 0.21938025210084033,
238
- "val_f1": 0.3448275822116528,
239
- "val_mae_log": 0.14887617528438568
240
- },
241
- {
242
- "epoch": 26,
243
- "train_loss": 0.04717193906371658,
244
- "val_loss": 0.012511309828947891,
245
- "val_auc_pr": 0.27487054503183533,
246
- "val_f1": 0.3636363596694215,
247
- "val_mae_log": 0.21023793518543243
248
- },
249
- {
250
- "epoch": 27,
251
- "train_loss": 0.04756235454130817,
252
- "val_loss": 0.01368828100914305,
253
- "val_auc_pr": 0.24682627084014383,
254
- "val_f1": 0.4444444409876544,
255
- "val_mae_log": 0.20768223702907562
256
- },
257
- {
258
- "epoch": 28,
259
- "train_loss": 0.04654277842995283,
260
- "val_loss": 0.01102768528190526,
261
- "val_auc_pr": 0.23941369620717445,
262
- "val_f1": 0.4444444409876544,
263
- "val_mae_log": 0.23168335855007172
264
- },
265
- {
266
- "epoch": 29,
267
- "train_loss": 0.04552056131934797,
268
- "val_loss": 0.010113749991763721,
269
- "val_auc_pr": 0.25363455988455985,
270
- "val_f1": 0.3888888854320988,
271
- "val_mae_log": 0.19058527052402496
272
- },
273
- {
274
- "epoch": 30,
275
- "train_loss": 0.046827327248615186,
276
- "val_loss": 0.012066270309415731,
277
- "val_auc_pr": 0.21310351047420012,
278
- "val_f1": 0.39024389929803693,
279
- "val_mae_log": 0.19954511523246765
280
- },
281
- {
282
- "epoch": 31,
283
- "train_loss": 0.04677234866933243,
284
- "val_loss": 0.01046423910355026,
285
- "val_auc_pr": 0.21076053433441952,
286
- "val_f1": 0.3529411728719723,
287
- "val_mae_log": 0.1768091470003128
288
- },
289
- {
290
- "epoch": 32,
291
- "train_loss": 0.04492179005331284,
292
- "val_loss": 0.010232081467455084,
293
- "val_auc_pr": 0.17865815250925546,
294
- "val_f1": 0.3333333302494332,
295
- "val_mae_log": 0.211782768368721
296
- },
297
- {
298
- "epoch": 33,
299
- "train_loss": 0.043496369318784894,
300
- "val_loss": 0.006482579978182912,
301
- "val_auc_pr": 0.28666645854145856,
302
- "val_f1": 0.4705882316955018,
303
- "val_mae_log": 0.12188047915697098
304
- },
305
- {
306
- "epoch": 34,
307
- "train_loss": 0.04519340708046346,
308
- "val_loss": 0.009889469291506843,
309
- "val_auc_pr": 0.27820205733808673,
310
- "val_f1": 0.4666666627555556,
311
- "val_mae_log": 0.1685512810945511
312
- },
313
- {
314
- "epoch": 35,
315
- "train_loss": 0.043441265519406344,
316
- "val_loss": 0.008723218459635973,
317
- "val_auc_pr": 0.2109856338037186,
318
- "val_f1": 0.3414634114931589,
319
- "val_mae_log": 0.166227787733078
320
- },
321
- {
322
- "epoch": 36,
323
- "train_loss": 0.044664783413345746,
324
- "val_loss": 0.006785003193230791,
325
- "val_auc_pr": 0.2637241077539003,
326
- "val_f1": 0.4285714244897959,
327
- "val_mae_log": 0.13594700396060944
328
- },
329
- {
330
- "epoch": 37,
331
- "train_loss": 0.0438888034506424,
332
- "val_loss": 0.00970885284583677,
333
- "val_auc_pr": 0.29875031687899806,
334
- "val_f1": 0.4374999962500001,
335
- "val_mae_log": 0.21538463234901428
336
- },
337
- {
338
- "epoch": 38,
339
- "train_loss": 0.04413037275945818,
340
- "val_loss": 0.005621075672520833,
341
- "val_auc_pr": 0.3619821967248438,
342
- "val_f1": 0.4705882316955018,
343
- "val_mae_log": 0.11580366641283035
344
- },
345
- {
346
- "epoch": 39,
347
- "train_loss": 0.04318939206769337,
348
- "val_loss": 0.009403676598925482,
349
- "val_auc_pr": 0.28100631598793363,
350
- "val_f1": 0.41666666222222226,
351
- "val_mae_log": 0.1686939150094986
352
- },
353
- {
354
- "epoch": 40,
355
- "train_loss": 0.04242932923339509,
356
- "val_loss": 0.009161080403084105,
357
- "val_auc_pr": 0.2549693362193362,
358
- "val_f1": 0.4210526282548477,
359
- "val_mae_log": 0.196553036570549
360
- },
361
- {
362
- "epoch": 41,
363
- "train_loss": 0.04452746674619817,
364
- "val_loss": 0.012993539429523727,
365
- "val_auc_pr": 0.41195570532527054,
366
- "val_f1": 0.45714285361632656,
367
- "val_mae_log": 0.26818016171455383
368
- },
369
- {
370
- "epoch": 42,
371
- "train_loss": 0.04150597550780386,
372
- "val_loss": 0.0069128969176249075,
373
- "val_auc_pr": 0.25035691011497463,
374
- "val_f1": 0.42424242056932976,
375
- "val_mae_log": 0.12961456179618835
376
- },
377
- {
378
- "epoch": 43,
379
- "train_loss": 0.04349473532591317,
380
- "val_loss": 0.006280094744976272,
381
- "val_auc_pr": 0.2996355184425671,
382
- "val_f1": 0.4374999962500001,
383
- "val_mae_log": 0.12820270657539368
384
- },
385
- {
386
- "epoch": 44,
387
- "train_loss": 0.042086653300636524,
388
- "val_loss": 0.006080356722866947,
389
- "val_auc_pr": 0.3077844154002621,
390
- "val_f1": 0.387096770364204,
391
- "val_mae_log": 0.1113549992442131
392
- },
393
- {
394
- "epoch": 45,
395
- "train_loss": 0.04128391085242903,
396
- "val_loss": 0.006863191321661527,
397
- "val_auc_pr": 0.2994744900595284,
398
- "val_f1": 0.4516128993964621,
399
- "val_mae_log": 0.1562904268503189
400
- },
401
- {
402
- "epoch": 46,
403
- "train_loss": 0.04144880730356719,
404
- "val_loss": 0.00688783231783997,
405
- "val_auc_pr": 0.3691604419545596,
406
- "val_f1": 0.4444444409876544,
407
- "val_mae_log": 0.1382068693637848
408
- },
409
- {
410
- "epoch": 47,
411
- "train_loss": 0.0430741350493721,
412
- "val_loss": 0.01172477337108417,
413
- "val_auc_pr": 0.354565090706395,
414
- "val_f1": 0.5161290284287201,
415
- "val_mae_log": 0.23300053179264069
416
- },
417
- {
418
- "epoch": 48,
419
- "train_loss": 0.04136262543096736,
420
- "val_loss": 0.006747952281412753,
421
- "val_auc_pr": 0.23154761904761906,
422
- "val_f1": 0.39999999680000003,
423
- "val_mae_log": 0.11889725923538208
424
- },
425
- {
426
- "epoch": 49,
427
- "train_loss": 0.04102441786310157,
428
- "val_loss": 0.005679253488779068,
429
- "val_auc_pr": 0.26123348724155177,
430
- "val_f1": 0.43478260415879016,
431
- "val_mae_log": 0.11081891506910324
432
- },
433
- {
434
- "epoch": 50,
435
- "train_loss": 0.041240745017657406,
436
- "val_loss": 0.008574311071160164,
437
- "val_auc_pr": 0.25022277917014757,
438
- "val_f1": 0.3999999964734694,
439
- "val_mae_log": 0.17056043446063995
440
- },
441
- {
442
- "epoch": 51,
443
- "train_loss": 0.04042079441652105,
444
- "val_loss": 0.006484658626670187,
445
- "val_auc_pr": 0.38022741147741146,
446
- "val_f1": 0.3999999950222222,
447
- "val_mae_log": 0.13463589549064636
448
- },
449
- {
450
- "epoch": 52,
451
- "train_loss": 0.040594170373436565,
452
- "val_loss": 0.007078204134648497,
453
- "val_auc_pr": 0.29691185040449747,
454
- "val_f1": 0.4374999962500001,
455
- "val_mae_log": 0.16555899381637573
456
- },
457
- {
458
- "epoch": 53,
459
- "train_loss": 0.03960525017936487,
460
- "val_loss": 0.004950580864467404,
461
- "val_auc_pr": 0.35856143069018487,
462
- "val_f1": 0.43478260415879016,
463
- "val_mae_log": 0.1040295735001564
464
- },
465
- {
466
- "epoch": 54,
467
- "train_loss": 0.03950474987158904,
468
- "val_loss": 0.005703899480233138,
469
- "val_auc_pr": 0.3339052143273096,
470
- "val_f1": 0.37499999500000003,
471
- "val_mae_log": 0.1224917396903038
472
- },
473
- {
474
- "epoch": 55,
475
- "train_loss": 0.03805918788184991,
476
- "val_loss": 0.005923548459329389,
477
- "val_auc_pr": 0.36686317155067155,
478
- "val_f1": 0.41666666222222226,
479
- "val_mae_log": 0.14333999156951904
480
- },
481
- {
482
- "epoch": 56,
483
- "train_loss": 0.03942116719444056,
484
- "val_loss": 0.004614247929897498,
485
- "val_auc_pr": 0.28647550366300367,
486
- "val_f1": 0.4285714244897959,
487
- "val_mae_log": 0.09474528580904007
488
- }
489
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
490
  }
 
1
+ {
2
+ "model": "PI-TFT (Physics-Informed Temporal Fusion Transformer)",
3
+ "best_epoch": 36,
4
+ "training_time_minutes": 13.175001474221547,
5
+ "optimal_threshold": 0.2639383375644684,
6
+ "temperature": 0.6179193258285522,
7
+ "use_density": true,
8
+ "test": {
9
+ "loss": 0.021245601093944383,
10
+ "auc_pr": 0.5076785607710974,
11
+ "auc_roc": 0.946749355627952,
12
+ "f1_at_50": 0.0,
13
+ "n_positive": 73,
14
+ "n_total": 2167,
15
+ "pos_rate": 0.03368712589144707,
16
+ "f1": 0.5185185137773299,
17
+ "optimal_threshold": 0.2639383375644684,
18
+ "threshold": 0.2639383375644684,
19
+ "recall_at_prec_30": 0.7808219178082192,
20
+ "recall_at_prec_50": 0.4931506849315068,
21
+ "recall_at_prec_70": 0.2876712328767123,
22
+ "mae_log": 0.10174570232629776,
23
+ "rmse_log": 0.15394317551905587,
24
+ "mae_km": 1533.616943359375,
25
+ "median_abs_error_km": 926.875
26
+ },
27
+ "test_calibrated": {
28
+ "auc_pr": 0.5076785607710974,
29
+ "auc_roc": 0.946749355627952,
30
+ "f1_at_50": 0.0,
31
+ "n_positive": 73,
32
+ "n_total": 2167,
33
+ "pos_rate": 0.03368712589144707,
34
+ "f1": 0.5185185137773299,
35
+ "optimal_threshold": 0.15979407727718353,
36
+ "threshold": 0.15979407727718353,
37
+ "recall_at_prec_30": 0.7808219178082192,
38
+ "recall_at_prec_50": 0.4931506849315068,
39
+ "recall_at_prec_70": 0.2876712328767123
40
+ },
41
+ "history": [
42
+ {
43
+ "epoch": 1,
44
+ "train_loss": 6.801268232190931,
45
+ "val_loss": 5.25680010659354,
46
+ "val_auc_pr": 0.007896529454622946,
47
+ "val_f1": 0.019323671026161646,
48
+ "val_mae_log": 7.11151123046875
49
+ },
50
+ {
51
+ "epoch": 2,
52
+ "train_loss": 3.834329532932591,
53
+ "val_loss": 2.7224787643977573,
54
+ "val_auc_pr": 0.010594418921027337,
55
+ "val_f1": 0.023529411193771638,
56
+ "val_mae_log": 5.041237831115723
57
+ },
58
+ {
59
+ "epoch": 3,
60
+ "train_loss": 1.955074778118649,
61
+ "val_loss": 1.1283516032355172,
62
+ "val_auc_pr": 0.008480584727743306,
63
+ "val_f1": 0.021505376131344667,
64
+ "val_mae_log": 3.112034797668457
65
+ },
66
+ {
67
+ "epoch": 4,
68
+ "train_loss": 0.6309667991625296,
69
+ "val_loss": 0.2000983421291624,
70
+ "val_auc_pr": 0.047413803659580166,
71
+ "val_f1": 0.11764705467128042,
72
+ "val_mae_log": 1.1961653232574463
73
+ },
74
+ {
75
+ "epoch": 5,
76
+ "train_loss": 0.13499785540877163,
77
+ "val_loss": 0.02656353052173342,
78
+ "val_auc_pr": 0.05766442486817594,
79
+ "val_f1": 0.15999999680000007,
80
+ "val_mae_log": 0.29869771003723145
81
+ },
82
+ {
83
+ "epoch": 6,
84
+ "train_loss": 0.07689017317182309,
85
+ "val_loss": 0.02750414184161595,
86
+ "val_auc_pr": 0.134885373440643,
87
+ "val_f1": 0.27272726921487606,
88
+ "val_mae_log": 0.3075650930404663
89
+ },
90
+ {
91
+ "epoch": 7,
92
+ "train_loss": 0.08175783813805193,
93
+ "val_loss": 0.07211375555821828,
94
+ "val_auc_pr": 0.18529914529914526,
95
+ "val_f1": 0.4285714239795918,
96
+ "val_mae_log": 0.6812126040458679
97
+ },
98
+ {
99
+ "epoch": 8,
100
+ "train_loss": 0.07750273872468923,
101
+ "val_loss": 0.027415024914911816,
102
+ "val_auc_pr": 0.13237697916045849,
103
+ "val_f1": 0.3157894698060942,
104
+ "val_mae_log": 0.35104697942733765
105
+ },
106
+ {
107
+ "epoch": 9,
108
+ "train_loss": 0.06653158048520218,
109
+ "val_loss": 0.01911477212394987,
110
+ "val_auc_pr": 0.20693184703085693,
111
+ "val_f1": 0.374999995703125,
112
+ "val_mae_log": 0.2960411608219147
113
+ },
114
+ {
115
+ "epoch": 10,
116
+ "train_loss": 0.0626621154917253,
117
+ "val_loss": 0.020604882389307022,
118
+ "val_auc_pr": 0.3348872180451128,
119
+ "val_f1": 0.5454545404958678,
120
+ "val_mae_log": 0.23688556253910065
121
+ },
122
+ {
123
+ "epoch": 11,
124
+ "train_loss": 0.0617836594581604,
125
+ "val_loss": 0.012763384197439467,
126
+ "val_auc_pr": 0.1294155844155844,
127
+ "val_f1": 0.22222221920438956,
128
+ "val_mae_log": 0.1817978173494339
129
+ },
130
+ {
131
+ "epoch": 12,
132
+ "train_loss": 0.05554375463240856,
133
+ "val_loss": 0.01185049262962171,
134
+ "val_auc_pr": 0.24263038548752833,
135
+ "val_f1": 0.36363635867768596,
136
+ "val_mae_log": 0.15147316455841064
137
+ },
138
+ {
139
+ "epoch": 13,
140
+ "train_loss": 0.05319682077781574,
141
+ "val_loss": 0.017937806567975452,
142
+ "val_auc_pr": 0.2786109128966272,
143
+ "val_f1": 0.33333333055555564,
144
+ "val_mae_log": 0.21772687137126923
145
+ },
146
+ {
147
+ "epoch": 14,
148
+ "train_loss": 0.05603743799634882,
149
+ "val_loss": 0.012255215285612004,
150
+ "val_auc_pr": 0.1654839208410637,
151
+ "val_f1": 0.3076923029585799,
152
+ "val_mae_log": 0.12889182567596436
153
+ },
154
+ {
155
+ "epoch": 15,
156
+ "train_loss": 0.052231158416818926,
157
+ "val_loss": 0.008827194571495056,
158
+ "val_auc_pr": 0.30569487983281085,
159
+ "val_f1": 0.4705882311418686,
160
+ "val_mae_log": 0.11871597170829773
161
+ },
162
+ {
163
+ "epoch": 16,
164
+ "train_loss": 0.050459702796227225,
165
+ "val_loss": 0.006688231070126806,
166
+ "val_auc_pr": 0.3174495864073329,
167
+ "val_f1": 0.33333333055555564,
168
+ "val_mae_log": 0.11670727282762527
169
+ },
170
+ {
171
+ "epoch": 17,
172
+ "train_loss": 0.05048987201943591,
173
+ "val_loss": 0.012136828287371568,
174
+ "val_auc_pr": 0.209023569023569,
175
+ "val_f1": 0.3529411723183391,
176
+ "val_mae_log": 0.15395033359527588
177
+ },
178
+ {
179
+ "epoch": 18,
180
+ "train_loss": 0.05087649694367035,
181
+ "val_loss": 0.007568871269800833,
182
+ "val_auc_pr": 0.2673856209150327,
183
+ "val_f1": 0.3999999962500001,
184
+ "val_mae_log": 0.1411171853542328
185
+ },
186
+ {
187
+ "epoch": 19,
188
+ "train_loss": 0.050642090935159374,
189
+ "val_loss": 0.0066412134495164666,
190
+ "val_auc_pr": 0.27475908192734455,
191
+ "val_f1": 0.3999999955555556,
192
+ "val_mae_log": 0.0915408581495285
193
+ },
194
+ {
195
+ "epoch": 20,
196
+ "train_loss": 0.04991532632628003,
197
+ "val_loss": 0.0055730888686542,
198
+ "val_auc_pr": 0.24940384615384617,
199
+ "val_f1": 0.33333332932098775,
200
+ "val_mae_log": 0.10347151011228561
201
+ },
202
+ {
203
+ "epoch": 21,
204
+ "train_loss": 0.049406778288854133,
205
+ "val_loss": 0.008397463309977735,
206
+ "val_auc_pr": 0.22877207681961503,
207
+ "val_f1": 0.2857142816326531,
208
+ "val_mae_log": 0.15620921552181244
209
+ },
210
+ {
211
+ "epoch": 22,
212
+ "train_loss": 0.04929839575008766,
213
+ "val_loss": 0.0075396452365177015,
214
+ "val_auc_pr": 0.3359158185268243,
215
+ "val_f1": 0.33333333055555564,
216
+ "val_mae_log": 0.11639901250600815
217
+ },
218
+ {
219
+ "epoch": 23,
220
+ "train_loss": 0.04896112705606061,
221
+ "val_loss": 0.007832049591732877,
222
+ "val_auc_pr": 0.3431446821152704,
223
+ "val_f1": 0.36363636012396694,
224
+ "val_mae_log": 0.10894307494163513
225
+ },
226
+ {
227
+ "epoch": 24,
228
+ "train_loss": 0.048813931744646384,
229
+ "val_loss": 0.0061542981836412635,
230
+ "val_auc_pr": 0.3559577677224736,
231
+ "val_f1": 0.36363636012396694,
232
+ "val_mae_log": 0.07847719639539719
233
+ },
234
+ {
235
+ "epoch": 25,
236
+ "train_loss": 0.04768835706888019,
237
+ "val_loss": 0.006223144009709358,
238
+ "val_auc_pr": 0.3659761291340239,
239
+ "val_f1": 0.421052627700831,
240
+ "val_mae_log": 0.14390207827091217
241
+ },
242
+ {
243
+ "epoch": 26,
244
+ "train_loss": 0.04840076712740434,
245
+ "val_loss": 0.0067752449374113765,
246
+ "val_auc_pr": 0.2586657651566374,
247
+ "val_f1": 0.34782608355387534,
248
+ "val_mae_log": 0.1449323147535324
249
+ },
250
+ {
251
+ "epoch": 27,
252
+ "train_loss": 0.047609428044509246,
253
+ "val_loss": 0.0065139371103474075,
254
+ "val_auc_pr": 0.34384112619406737,
255
+ "val_f1": 0.34782608355387534,
256
+ "val_mae_log": 0.09073375165462494
257
+ },
258
+ {
259
+ "epoch": 28,
260
+ "train_loss": 0.04662630880201185,
261
+ "val_loss": 0.006256445976240295,
262
+ "val_auc_pr": 0.33832141293241863,
263
+ "val_f1": 0.33333333055555564,
264
+ "val_mae_log": 0.07596895098686218
265
+ },
266
+ {
267
+ "epoch": 29,
268
+ "train_loss": 0.04634691820152708,
269
+ "val_loss": 0.005017333896830678,
270
+ "val_auc_pr": 0.336514012303486,
271
+ "val_f1": 0.33333333055555564,
272
+ "val_mae_log": 0.07677556574344635
273
+ },
274
+ {
275
+ "epoch": 30,
276
+ "train_loss": 0.04663669626052315,
277
+ "val_loss": 0.004762223763723991,
278
+ "val_auc_pr": 0.24682988580047405,
279
+ "val_f1": 0.36363636012396694,
280
+ "val_mae_log": 0.08992886543273926
281
+ },
282
+ {
283
+ "epoch": 31,
284
+ "train_loss": 0.046282403110652355,
285
+ "val_loss": 0.003826435888186097,
286
+ "val_auc_pr": 0.2284485407066052,
287
+ "val_f1": 0.3999999962500001,
288
+ "val_mae_log": 0.06141701713204384
289
+ },
290
+ {
291
+ "epoch": 32,
292
+ "train_loss": 0.04575154318197353,
293
+ "val_loss": 0.005115043604746461,
294
+ "val_auc_pr": 0.3611255411255411,
295
+ "val_f1": 0.3999999962500001,
296
+ "val_mae_log": 0.09008380770683289
297
+ },
298
+ {
299
+ "epoch": 33,
300
+ "train_loss": 0.046043931763317135,
301
+ "val_loss": 0.004483342935730304,
302
+ "val_auc_pr": 0.36333333333333334,
303
+ "val_f1": 0.3809523773242631,
304
+ "val_mae_log": 0.10232321172952652
305
+ },
306
+ {
307
+ "epoch": 34,
308
+ "train_loss": 0.04492839058307377,
309
+ "val_loss": 0.007276699944798436,
310
+ "val_auc_pr": 0.3461904761904762,
311
+ "val_f1": 0.3809523773242631,
312
+ "val_mae_log": 0.10686437785625458
313
+ },
314
+ {
315
+ "epoch": 35,
316
+ "train_loss": 0.04576677558188503,
317
+ "val_loss": 0.004259714224774923,
318
+ "val_auc_pr": 0.37718954248366016,
319
+ "val_f1": 0.3999999962500001,
320
+ "val_mae_log": 0.0769796371459961
321
+ },
322
+ {
323
+ "epoch": 36,
324
+ "train_loss": 0.044130372638637956,
325
+ "val_loss": 0.004274079659288483,
326
+ "val_auc_pr": 0.4215151515151515,
327
+ "val_f1": 0.4444444395061729,
328
+ "val_mae_log": 0.09318451583385468
329
+ },
330
+ {
331
+ "epoch": 37,
332
+ "train_loss": 0.04556343443691731,
333
+ "val_loss": 0.0053521015548280305,
334
+ "val_auc_pr": 0.3828373015873016,
335
+ "val_f1": 0.421052627700831,
336
+ "val_mae_log": 0.11446798592805862
337
+ },
338
+ {
339
+ "epoch": 38,
340
+ "train_loss": 0.04497031863476779,
341
+ "val_loss": 0.005016647595246988,
342
+ "val_auc_pr": 0.38186813186813184,
343
+ "val_f1": 0.3809523773242631,
344
+ "val_mae_log": 0.11497646570205688
345
+ },
346
+ {
347
+ "epoch": 39,
348
+ "train_loss": 0.04312905277553442,
349
+ "val_loss": 0.003749881671475513,
350
+ "val_auc_pr": 0.3595238095238095,
351
+ "val_f1": 0.3809523773242631,
352
+ "val_mae_log": 0.05548140034079552
353
+ },
354
+ {
355
+ "epoch": 40,
356
+ "train_loss": 0.04352163130769859,
357
+ "val_loss": 0.005372332009885993,
358
+ "val_auc_pr": 0.3503288825869471,
359
+ "val_f1": 0.34782608355387534,
360
+ "val_mae_log": 0.08230870962142944
361
+ },
362
+ {
363
+ "epoch": 41,
364
+ "train_loss": 0.043740846146200156,
365
+ "val_loss": 0.0039979582319834405,
366
+ "val_auc_pr": 0.41458333333333336,
367
+ "val_f1": 0.3999999962500001,
368
+ "val_mae_log": 0.08734633028507233
369
+ },
370
+ {
371
+ "epoch": 42,
372
+ "train_loss": 0.04409235781310378,
373
+ "val_loss": 0.005109895303446267,
374
+ "val_auc_pr": 0.2524756335282651,
375
+ "val_f1": 0.33333333003472226,
376
+ "val_mae_log": 0.07870446890592575
377
+ },
378
+ {
379
+ "epoch": 43,
380
+ "train_loss": 0.043179894389735685,
381
+ "val_loss": 0.005041864268215639,
382
+ "val_auc_pr": 0.26508912655971484,
383
+ "val_f1": 0.36363636012396694,
384
+ "val_mae_log": 0.07578516006469727
385
+ },
386
+ {
387
+ "epoch": 44,
388
+ "train_loss": 0.04234155755792115,
389
+ "val_loss": 0.0038543779269925187,
390
+ "val_auc_pr": 0.3427519893899204,
391
+ "val_f1": 0.33333333055555564,
392
+ "val_mae_log": 0.06378159672021866
393
+ },
394
+ {
395
+ "epoch": 45,
396
+ "train_loss": 0.043199574021068776,
397
+ "val_loss": 0.00448337330349854,
398
+ "val_auc_pr": 0.38693977591036416,
399
+ "val_f1": 0.36363636012396694,
400
+ "val_mae_log": 0.08112290501594543
401
+ },
402
+ {
403
+ "epoch": 46,
404
+ "train_loss": 0.04324697579282361,
405
+ "val_loss": 0.004593804511906845,
406
+ "val_auc_pr": 0.3657142857142857,
407
+ "val_f1": 0.3809523773242631,
408
+ "val_mae_log": 0.12126877903938293
409
+ },
410
+ {
411
+ "epoch": 47,
412
+ "train_loss": 0.042983541144309814,
413
+ "val_loss": 0.0034202520120223717,
414
+ "val_auc_pr": 0.36703703703703705,
415
+ "val_f1": 0.3809523773242631,
416
+ "val_mae_log": 0.05318637564778328
417
+ },
418
+ {
419
+ "epoch": 48,
420
+ "train_loss": 0.04088504479543583,
421
+ "val_loss": 0.0037384599480511887,
422
+ "val_auc_pr": 0.35812684047978166,
423
+ "val_f1": 0.38461538150887575,
424
+ "val_mae_log": 0.0607416033744812
425
+ },
426
+ {
427
+ "epoch": 49,
428
+ "train_loss": 0.0411647165143812,
429
+ "val_loss": 0.0038923417118244936,
430
+ "val_auc_pr": 0.37444444444444447,
431
+ "val_f1": 0.3809523773242631,
432
+ "val_mae_log": 0.07454186677932739
433
+ },
434
+ {
435
+ "epoch": 50,
436
+ "train_loss": 0.04235347539589212,
437
+ "val_loss": 0.0035431724141484927,
438
+ "val_auc_pr": 0.3718181818181818,
439
+ "val_f1": 0.3809523773242631,
440
+ "val_mae_log": 0.05186235159635544
441
+ },
442
+ {
443
+ "epoch": 51,
444
+ "train_loss": 0.03975096909782371,
445
+ "val_loss": 0.003855357279202768,
446
+ "val_auc_pr": 0.37,
447
+ "val_f1": 0.3809523773242631,
448
+ "val_mae_log": 0.08433445543050766
449
+ },
450
+ {
451
+ "epoch": 52,
452
+ "train_loss": 0.040304526777283564,
453
+ "val_loss": 0.003954493274380054,
454
+ "val_auc_pr": 0.36705882352941177,
455
+ "val_f1": 0.36363636012396694,
456
+ "val_mae_log": 0.0650041252374649
457
+ },
458
+ {
459
+ "epoch": 53,
460
+ "train_loss": 0.041316902365636184,
461
+ "val_loss": 0.0044658422370308214,
462
+ "val_auc_pr": 0.37444444444444447,
463
+ "val_f1": 0.39999999680000003,
464
+ "val_mae_log": 0.08514165133237839
465
+ },
466
+ {
467
+ "epoch": 54,
468
+ "train_loss": 0.041085500773545856,
469
+ "val_loss": 0.003584100299381784,
470
+ "val_auc_pr": 0.36991596638655466,
471
+ "val_f1": 0.36363636012396694,
472
+ "val_mae_log": 0.04943912476301193
473
+ },
474
+ {
475
+ "epoch": 55,
476
+ "train_loss": 0.04048956327543066,
477
+ "val_loss": 0.003669723236401166,
478
+ "val_auc_pr": 0.366961926961927,
479
+ "val_f1": 0.34782608355387534,
480
+ "val_mae_log": 0.0743192732334137
481
+ },
482
+ {
483
+ "epoch": 56,
484
+ "train_loss": 0.04016674624101536,
485
+ "val_loss": 0.004304527521266469,
486
+ "val_auc_pr": 0.3745588235294118,
487
+ "val_f1": 0.39999999680000003,
488
+ "val_mae_log": 0.08440288156270981
489
+ }
490
+ ],
491
+ "conformal": {
492
+ "alpha_0.01": {
493
+ "conformal_metrics": {
494
+ "alpha": 0.01,
495
+ "target_coverage": 0.99,
496
+ "marginal_coverage": 0.9700046146746655,
497
+ "coverage_guarantee_met": false,
498
+ "avg_set_size": 2.1033687125057683,
499
+ "efficiency": 0.4741578218735579,
500
+ "positive_coverage": 0.136986301369863,
501
+ "negative_coverage": 0.9990448901623686,
502
+ "set_size_distribution": {
503
+ "2": 1948,
504
+ "3": 214,
505
+ "4": 5
506
+ },
507
+ "n_test": 2167,
508
+ "mean_interval_width": 0.35249775648117065,
509
+ "median_interval_width": 0.3299492597579956
510
+ },
511
+ "conformal_state": {
512
+ "is_calibrated": true,
513
+ "alpha": 0.01,
514
+ "q_hat": 0.31530878875241947,
515
+ "q_residual": 0.31530878875241947,
516
+ "n_cal": 527,
517
+ "tiers": {
518
+ "LOW": [
519
+ 0.0,
520
+ 0.1
521
+ ],
522
+ "MODERATE": [
523
+ 0.1,
524
+ 0.4
525
+ ],
526
+ "HIGH": [
527
+ 0.4,
528
+ 0.7
529
+ ],
530
+ "CRITICAL": [
531
+ 0.7,
532
+ 1.0
533
+ ]
534
+ }
535
+ }
536
+ },
537
+ "alpha_0.05": {
538
+ "conformal_metrics": {
539
+ "alpha": 0.05,
540
+ "target_coverage": 0.95,
541
+ "marginal_coverage": 0.9487771112136595,
542
+ "coverage_guarantee_met": true,
543
+ "avg_set_size": 1.9856945085371482,
544
+ "efficiency": 0.503576372865713,
545
+ "positive_coverage": 0.0,
546
+ "negative_coverage": 0.9818529130850048,
547
+ "set_size_distribution": {
548
+ "1": 31,
549
+ "2": 2136
550
+ },
551
+ "n_test": 2167,
552
+ "mean_interval_width": 0.14139389991760254,
553
+ "median_interval_width": 0.1266784965991974
554
+ },
555
+ "conformal_state": {
556
+ "is_calibrated": true,
557
+ "alpha": 0.05,
558
+ "q_hat": 0.1120380280677236,
559
+ "q_residual": 0.1120380280677236,
560
+ "n_cal": 527,
561
+ "tiers": {
562
+ "LOW": [
563
+ 0.0,
564
+ 0.1
565
+ ],
566
+ "MODERATE": [
567
+ 0.1,
568
+ 0.4
569
+ ],
570
+ "HIGH": [
571
+ 0.4,
572
+ 0.7
573
+ ],
574
+ "CRITICAL": [
575
+ 0.7,
576
+ 1.0
577
+ ]
578
+ }
579
+ }
580
+ },
581
+ "alpha_0.1": {
582
+ "conformal_metrics": {
583
+ "alpha": 0.1,
584
+ "target_coverage": 0.9,
585
+ "marginal_coverage": 0.9284725426857406,
586
+ "coverage_guarantee_met": true,
587
+ "avg_set_size": 1.103830179972312,
588
+ "efficiency": 0.724042455006922,
589
+ "positive_coverage": 0.0,
590
+ "negative_coverage": 0.9608404966571156,
591
+ "set_size_distribution": {
592
+ "1": 1942,
593
+ "2": 225
594
+ },
595
+ "n_test": 2167,
596
+ "mean_interval_width": 0.060726769268512726,
597
+ "median_interval_width": 0.05510023236274719
598
+ },
599
+ "conformal_state": {
600
+ "is_calibrated": true,
601
+ "alpha": 0.1,
602
+ "q_hat": 0.04045976169647709,
603
+ "q_residual": 0.04045976169647709,
604
+ "n_cal": 527,
605
+ "tiers": {
606
+ "LOW": [
607
+ 0.0,
608
+ 0.1
609
+ ],
610
+ "MODERATE": [
611
+ 0.1,
612
+ 0.4
613
+ ],
614
+ "HIGH": [
615
+ 0.4,
616
+ 0.7
617
+ ],
618
+ "CRITICAL": [
619
+ 0.7,
620
+ 1.0
621
+ ]
622
+ }
623
+ }
624
+ },
625
+ "alpha_0.2": {
626
+ "conformal_metrics": {
627
+ "alpha": 0.2,
628
+ "target_coverage": 0.8,
629
+ "marginal_coverage": 0.9220119981541302,
630
+ "coverage_guarantee_met": true,
631
+ "avg_set_size": 1.054453161052146,
632
+ "efficiency": 0.7363867097369635,
633
+ "positive_coverage": 0.0,
634
+ "negative_coverage": 0.9541547277936963,
635
+ "set_size_distribution": {
636
+ "1": 2049,
637
+ "2": 118
638
+ },
639
+ "n_test": 2167,
640
+ "mean_interval_width": 0.04071307182312012,
641
+ "median_interval_width": 0.039181869477033615
642
+ },
643
+ "conformal_state": {
644
+ "is_calibrated": true,
645
+ "alpha": 0.2,
646
+ "q_hat": 0.024541400479014954,
647
+ "q_residual": 0.024541400479014954,
648
+ "n_cal": 527,
649
+ "tiers": {
650
+ "LOW": [
651
+ 0.0,
652
+ 0.1
653
+ ],
654
+ "MODERATE": [
655
+ 0.1,
656
+ 0.4
657
+ ],
658
+ "HIGH": [
659
+ 0.4,
660
+ 0.7
661
+ ],
662
+ "CRITICAL": [
663
+ 0.7,
664
+ 1.0
665
+ ]
666
+ }
667
+ }
668
+ }
669
+ }
670
  }