add informer dataset classes; fix configs

state-spaces · Jan 30, 2022 · cfb6a06 · cfb6a06
1 parent feeab74
commit cfb6a06
Show file tree

Hide file tree

Showing 6 changed files with 636 additions and 5 deletions.
diff --git a/configs/experiment/s4-informer-ecl.yaml b/configs/experiment/s4-informer-ecl.yaml
@@ -2,7 +2,7 @@
 defaults:
   - /pipeline: informer
   - /model: s4
-  - /dataset: ecl
+  - override /dataset: ecl
 
 trainer:
   max_epochs: 10

diff --git a/configs/experiment/s4-informer-etth.yaml b/configs/experiment/s4-informer-etth.yaml
@@ -2,7 +2,7 @@
 defaults:
   - /pipeline: informer
   - /model: s4
-  - /dataset: etth
+  - override /dataset: etth
 
 trainer:
   max_epochs: 10

diff --git a/configs/experiment/s4-informer-ettm.yaml b/configs/experiment/s4-informer-ettm.yaml
@@ -2,7 +2,7 @@
 defaults:
   - /pipeline: informer
   - /model: s4
-  - /dataset: ettm
+  - override /dataset: ettm
 
 trainer:
   max_epochs: 10

diff --git a/configs/experiment/s4-informer-weather.yaml b/configs/experiment/s4-informer-weather.yaml
@@ -2,7 +2,7 @@
 defaults:
   - /pipeline: informer
   - /model: s4
-  - /dataset: weather
+  - override /dataset: weather
 
 trainer:
   max_epochs: 10

diff --git a/src/dataloaders/__init__.py b/src/dataloaders/__init__.py
@@ -1,2 +1,2 @@
 from .datasets import SequenceDataset
-from . import lm
+from . import lm, et