tests: 📏 add smoke tests for classification and imputation (#288)

Author: duyifanict

tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/meta.json

@@ -0,0 +1,39 @@
+{
+    "name": "ArticularyWordRecognition_mini",
+    "num_classes": 25,
+    "class_names": [
+        "1.0",
+        "10.0",
+        "11.0",
+        "12.0",
+        "13.0",
+        "14.0",
+        "15.0",
+        "16.0",
+        "17.0",
+        "18.0",
+        "19.0",
+        "2.0",
+        "20.0",
+        "21.0",
+        "22.0",
+        "23.0",
+        "24.0",
+        "25.0",
+        "3.0",
+        "4.0",
+        "5.0",
+        "6.0",
+        "7.0",
+        "8.0",
+        "9.0"
+    ],
+    "equal_length": true,
+    "seq_len": 144,
+    "num_nodes": 9,
+    "num_features": 1,
+    "shape": "[num_samples, seq_len, num_nodes, num_features]",
+    "missing": false,
+    "filling_missing": "NA",
+    "norm_each_channel": true
+}

tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_inputs.npy

@@ -0,0 +1,34 @@
+# pylint: disable=wrong-import-position
+
+import os
+import sys
+
+sys.path.append(os.path.abspath(__file__ + "/../../../src/"))
+os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__))))
+
+from basicts import BasicTSLauncher
+from basicts.configs import BasicTSClassificationConfig
+from basicts.models.iTransformer import (iTransformerConfig,
+                                         iTransformerForClassification)
+
+
+def test_itransformerforc_smoke_test():
+
+    model_config = iTransformerConfig(
+        input_len=144,
+        num_features=9,
+        num_classes=25
+        )
+
+    BasicTSLauncher.launch_training(BasicTSClassificationConfig(
+        model=iTransformerForClassification,
+        model_config=model_config,
+        dataset_name="ArticularyWordRecognition_mini",
+        gpus=None,
+        batch_size=16,
+        num_epochs=5,
+    ))
+
+
+if __name__ == "__main__":
+    test_itransformerforc_smoke_test()

tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_labels.npy

@@ -0,0 +1,35 @@
+# pylint: disable=wrong-import-position
+
+import os
+import sys
+
+sys.path.append(os.path.abspath(__file__ + "/../../../src/"))
+os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__))))
+
+from basicts import BasicTSLauncher
+from basicts.configs import BasicTSImputationConfig
+from basicts.models.iTransformer import (iTransformerConfig,
+                                         iTransformerForReconstruction)
+
+
+def test_itransformerforr_smoke_test():
+    input_len=32
+    model_config = iTransformerConfig(
+        input_len=input_len,
+        num_features=7
+        )
+
+    BasicTSLauncher.launch_training(BasicTSImputationConfig(
+        model=iTransformerForReconstruction,
+        model_config=model_config,
+        dataset_name="ETTh1_mini",
+        mask_ratio=0.25,
+        gpus=None,
+        batch_size=16,
+        input_len=input_len,
+        num_epochs=5,
+    ))
+
+
+if __name__ == "__main__":
+    test_itransformerforr_smoke_test()