diff --git a/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/meta.json b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/meta.json new file mode 100644 index 00000000..915e895f --- /dev/null +++ b/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 +} \ No newline at end of file diff --git a/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_inputs.npy b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_inputs.npy new file mode 100644 index 00000000..8f7a4a81 Binary files /dev/null and b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_inputs.npy differ diff --git a/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_labels.npy b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_labels.npy new file mode 100644 index 00000000..92ff2031 Binary files /dev/null and b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/test_labels.npy differ diff --git a/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/train_inputs.npy b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/train_inputs.npy new file mode 100644 index 00000000..f3420ec1 Binary files /dev/null and b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/train_inputs.npy differ diff --git a/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/train_labels.npy b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/train_labels.npy new file mode 100644 index 00000000..a16331f5 Binary files /dev/null and b/tests/smoke_test/datasets/UEA/ArticularyWordRecognition_mini/train_labels.npy differ diff --git a/tests/smoke_test/test_itransformerforc.py b/tests/smoke_test/test_itransformerforc.py new file mode 100644 index 00000000..022351f1 --- /dev/null +++ b/tests/smoke_test/test_itransformerforc.py @@ -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() diff --git a/tests/smoke_test/test_itransformerforr.py b/tests/smoke_test/test_itransformerforr.py new file mode 100644 index 00000000..03bc353b --- /dev/null +++ b/tests/smoke_test/test_itransformerforr.py @@ -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()