update

example/results/config_used.json

@@ -33,18 +33,18 @@
   "model_pos_dim": 64,
   "model_use_pos_embed": true,
   "disc_mask_scale": 0.9,
-  "shuffle_buffer": 1024,
+  "shuffle_buffer": 256,
   "cont_loss_weighting": "inv_std",
   "cont_loss_eps": 1e-06,
   "cont_target": "x0",
   "cont_clamp_x0": 5.0,
   "use_temporal_stage1": true,
-  "temporal_hidden_dim": 512,
+  "temporal_hidden_dim": 256,
-  "temporal_num_layers": 2,
+  "temporal_num_layers": 1,
   "temporal_dropout": 0.0,
-  "temporal_epochs": 5,
+  "temporal_epochs": 2,
-  "temporal_lr": 0.0005,
+  "temporal_lr": 0.001,
-  "quantile_loss_weight": 0.0,
+  "quantile_loss_weight": 0.2,
   "quantile_points": [
     0.05,
     0.25,
@@ -54,12 +54,15 @@
   ],
   "snr_weighted_loss": true,
   "snr_gamma": 1.0,
-  "residual_stat_weight": 0.0,
+  "residual_stat_weight": 0.05,
   "backbone_type": "transformer",
   "transformer_num_layers": 2,
   "transformer_nhead": 4,
   "transformer_ff_dim": 512,
   "transformer_dropout": 0.1,
+  "use_quantile_transform": true,
+  "quantile_bins": 1001,
+  "cont_bound_mode": "sigmoid",
   "sample_batch_size": 4,
   "sample_seq_len": 128
 }

example/results/metrics_history.csv

@@ -3,3 +3,5 @@ timestamp,avg_ks,avg_jsd,avg_lag1_diff
 2026-01-26T17:15:06.977813,0.5204744693396227,0.013087877218697579,0.6060810896277049
 2026-01-26T17:23:19.986340,0.6451295400943398,0.027818187919735473,0.24898834978597875
 2026-01-26T17:23:20.046864,0.6451295400943398,0.027818187919735473,0.24898834978597875
+2026-01-27T11:19:53.735830,0.9431816627358489,0.11505222210663352,0.39121478044554636
+2026-01-27T11:19:53.788440,0.9431816627358489,0.11505222210663352,0.39121478044554636

example/results/train_log.csv

@@ -1,61 +1,61 @@
 epoch,step,loss,loss_cont,loss_disc
-0,0,5523.104980,7889.408203,1.731906
+0,0,184976.187500,139257.531250,1.765069
-0,10,3276.585205,4680.372559,1.081256
+0,10,154583.593750,93562.953125,1.065642
-0,20,3275.231201,4678.548828,0.823748
+0,20,123729.125000,48789.566406,0.940561
-0,30,2829.968018,4042.515137,0.691539
+0,30,128532.953125,55698.578125,0.962107
-0,40,2078.723389,2969.359375,0.573009
+0,40,114552.265625,36245.523438,0.843481
-0,50,1930.569702,2757.771240,0.432756
+0,50,110819.375000,28501.634766,0.833165
-1,0,2158.404541,3083.257568,0.414602
+1,0,113179.671875,36839.066406,0.873785
-1,10,1826.730103,2609.486328,0.299258
+1,10,108036.046875,27230.572266,0.795820
-1,20,1948.610840,2783.595215,0.314316
+1,20,114723.328125,36305.980469,0.669550
-1,30,1750.570068,2500.690186,0.289544
+1,30,116334.398438,38525.492188,0.754736
-1,40,1797.315430,2567.464111,0.302009
+1,40,101621.039062,17570.927734,0.639032
-1,50,1803.055908,2575.670898,0.287583
+1,50,101044.445312,14520.701172,0.672534
-2,0,1742.254639,2488.802979,0.308467
+2,0,106989.359375,28113.287109,0.748860
-2,10,1680.838379,2401.103271,0.220643
+2,10,98382.351562,13329.769531,0.700875
-2,20,1676.041748,2394.241455,0.242607
+2,20,100775.515625,16308.768555,0.539620
-2,30,1672.020874,2388.488281,0.263532
+2,30,96904.796875,10761.526367,0.563072
-2,40,1684.479858,2406.280273,0.279219
+2,40,100871.171875,16531.033203,0.511787
-2,50,1694.152100,2420.096191,0.282612
+2,50,95249.234375,6386.864746,0.519408
-3,0,1675.018799,2392.732422,0.353447
+3,0,91759.843750,6220.618652,0.652666
-3,10,1653.036621,2361.412842,0.158843
+3,10,93521.960938,6607.199707,0.491660
-3,20,1642.358887,2346.104004,0.286732
+3,20,100551.515625,15872.579102,0.463317
-3,30,1605.379395,2293.295410,0.241910
+3,30,96029.742188,9569.395508,0.501565
-3,40,1629.166504,2327.272217,0.253239
+3,40,94091.968750,6877.456055,0.442239
-3,50,1629.753418,2328.095703,0.288231
+3,50,97826.546875,9955.442383,0.446910
-4,0,1671.573730,2387.815186,0.343630
+4,0,91900.179688,6396.767578,0.598208
-4,10,1613.391846,2304.772705,0.170222
+4,10,93740.500000,6541.960449,0.426235
-4,20,1609.151855,2298.696289,0.214823
+4,20,92498.335938,4718.511719,0.358245
-4,30,1620.748169,2315.245850,0.253361
+4,30,92927.234375,5127.920898,0.405424
-4,40,1656.549438,2366.396484,0.239846
+4,40,93010.304688,5518.281250,0.366179
-4,50,1550.946045,2215.520264,0.273209
+4,50,93015.593750,3258.701904,0.349383
-5,0,1714.970215,2449.819336,0.322403
+5,0,94799.273438,10739.668945,0.554139
-5,10,1617.331787,2310.389160,0.197981
+5,10,97879.421875,12976.933594,0.437371
-5,20,1616.368408,2309.002686,0.221823
+5,20,95471.609375,8845.887695,0.327026
-5,30,1560.830933,2229.644287,0.266509
+5,30,92265.351562,4189.309082,0.377223
-5,40,1576.884155,2252.579590,0.261611
+5,40,93774.914062,6497.603027,0.320435
-5,50,1589.115723,2270.042969,0.285631
+5,50,96035.773438,7816.457520,0.331506
-6,0,1607.691040,2296.556396,0.338550
+6,0,94584.390625,10466.210938,0.542753
-6,10,1559.538818,2227.814453,0.229090
+6,10,94790.671875,8768.079102,0.352866
-6,20,1583.750366,2262.401367,0.231597
+6,20,93085.093750,5598.226074,0.278421
-6,30,1569.908936,2242.621582,0.246370
+6,30,99669.890625,14897.969727,0.349784
-6,40,1492.759521,2132.401855,0.260828
+6,40,92761.164062,5166.158691,0.309837
-6,50,1593.370728,2276.132324,0.260473
+6,50,96724.195312,8688.732422,0.300651
-7,0,1631.732178,2330.911865,0.312931
+7,0,90174.679688,4375.904297,0.572326
-7,10,1646.043823,2351.406006,0.199128
+7,10,94914.375000,8772.841797,0.366727
-7,20,1562.786621,2232.436035,0.271245
+7,20,92351.898438,4551.166992,0.278914
-7,30,1524.693481,2178.024414,0.254854
+7,30,94377.937500,7232.752930,0.314258
-7,40,1541.151367,2201.529053,0.270027
+7,40,99633.523438,15073.370117,0.303307
-7,50,1491.367188,2130.424316,0.233909
+7,50,94508.851562,5760.681152,0.341641
-8,0,1575.110474,2250.026367,0.307003
+8,0,92399.125000,7706.516113,0.612074
-8,10,1546.139526,2208.707275,0.148086
+8,10,91553.718750,4061.810303,0.326436
-8,20,1581.709106,2259.487061,0.227342
+8,20,92247.484375,4653.951172,0.248402
-8,30,1534.755737,2192.393066,0.268629
+8,30,95060.500000,8586.428711,0.324341
-8,40,1506.180908,2151.576416,0.257973
+8,40,91353.234375,3453.355225,0.312784
-8,50,1527.244019,2181.677246,0.233319
+8,50,94687.640625,5992.084961,0.281537
-9,0,1585.274536,2264.521973,0.363625
+9,0,89032.601562,2830.959229,0.524878
-9,10,1629.692261,2328.056152,0.176398
+9,10,91304.429688,3972.131592,0.338050
-9,20,1537.658691,2196.561768,0.218023
+9,20,95922.851562,10126.095703,0.249572
-9,30,1511.831299,2159.663574,0.222698
+9,30,94574.312500,7888.812988,0.325134
-9,40,1492.001831,2131.322998,0.252746
+9,40,93815.937500,7084.785156,0.303094
-9,50,1606.072144,2294.288818,0.233116
+9,50,94993.406250,6770.722168,0.312788