Downloads · 30 days
0
SuryanshSS1011/match-loss-to-cost-checkpoints
match-loss-to-cost-checkpoints is a machine learning model from SuryanshSS1011. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-4.0.
These are example end-of-run model weights, one per architecture: LSTM, DLinear, PatchTST, iTransformer, and DCRNN. They are reference checkpoints rather than a per-seed canonical set, because the training pipeline ov…
Downloads · 30 days
0
Access
Public
Updated May 31, 2026
Repo size
8.9 MB
Likes
0
Public
Click a slice to open those files.
.pt4.3 MB · 100%
From the Hugging Face model README
These are example end-of-run model weights, one per architecture: LSTM,
DLinear, PatchTST, iTransformer, and DCRNN. They are reference
checkpoints rather than a per-seed canonical set, because the training
pipeline overwrites models/ on each run, so each file is just the
last training of that architecture from whichever sweep finished most
recently. Treat them as illustrative.
For the canonical reproduction path, use the predictions dataset at
SuryanshSS1011/match-loss-to-cost-predictions,
which regenerates every paper table and figure without retraining. To
get a fresh checkpoint matched to a specific seed, retrain from the
seed list documented in the code repo at
https://github.com/SuryanshSS1011/match-loss-to-cost.
The headline experiments use the LSTM backbone, and the other
architectures are loss-ablation backbones referenced in §5.C of the
paper. Each .pt is a plain PyTorch state_dict loadable by the
corresponding model class in src/models/ of the code repo.