Commit Graph

4 Commits

Author SHA1 Message Date
ef307a57e9 feat: add mamba and dynamic chunking related code and test code 2025-09-04 01:32:13 +00:00
12cb7652cf feat: add sigmoid learning rate scheduler
Add new 'sigmoid' learning rate adjustment method to utils/tools.py
with logistic growth rate and warm-up smoothing for better training
convergence.
2025-09-03 07:13:25 +00:00
a069c9a874 feat: add PEMS and Solar dataset support
- Add Dataset_PEMS and Dataset_Solar classes for PEMS and Solar datasets
- Update data_factory.py to include new dataset mappings
- Fix M4 dataset handling with proper numpy array dtype
- Add PEMS-specific loss function (L1Loss) and inverse transform support
- Update validation logic for PEMS dataset with inverse scaling
- Fix M4 data loader insample mask calculation bug

Changes support new traffic and solar energy datasets while maintaining
backward compatibility with existing datasets.
2025-09-03 07:01:32 +00:00
d6dd462886 first commit 2025-08-28 10:17:59 +00:00