feat: 🎸 add Timer and fix some bugs

Author: yisongfu

basicts/benchmarks/time_series_forecasting/ETTh1/itransformer.py

@@ -22,17 +22,15 @@ def iTransformer_ETTh1(
                     and training_config.dataset_name == "ETTh1", \
                     "The model should be iTransformer and dataset should be ETTh1."
             else:
-                model = iTransformerForForecasting(
-                    iTransformerConfig(
+                training_config = BasicTSForecastingConfig(
+                    model=iTransformerForForecasting,
+                    model_config=iTransformerConfig(
                         input_len=input_len,
                         output_len=output_len,
                         num_features=7,
                         hidden_size=32,
                         intermediate_size=32
-                    )
-                )
-                training_config = BasicTSForecastingConfig(
-                    model=model,
+                    ),
                     dataset_name="ETTh1",
                     input_len=input_len,
                     output_len=output_len,

basicts/configs/__init__.py

@@ -1,9 +1,13 @@
 from .base_config import BasicTSConfig
-from .forecasting_config import BasicTSForecastingConfig
 from .model_config import BasicTSModelConfig
+from .tsc_config import BasicTSClassificationConfig
+from .tsf_config import BasicTSForecastingConfig
 from .tsfm_config import BasicTSFoundationModelConfig
+from .tsi_config import BasicTSImputationConfig
 
 __ALL__ = ['BasicTSConfig',
            'BasicTSForecastingConfig',
+           'BasicTSClassificationConfig',
+           'BasicTSImputationConfig',
            'BasicTSFoundationModelConfig',
            'BasicTSModelConfig']

basicts/configs/base_config.py

@@ -17,6 +17,8 @@
 from basicts.runners.callback import BasicTSCallback
 from basicts.runners.taskflow import BasicTSTaskFlow
 
+from .model_config import BasicTSModelConfig
+
 
 @dataclass
 class BasicTSConfig(EasyDict):
@@ -28,7 +30,9 @@ class BasicTSConfig(EasyDict):
         d (dict, optional): Dictionary to initialize the configuration. Defaults to None.
     """
 
-    model: torch.nn.Module
+    model: type
+    model_config: BasicTSModelConfig
+
     dataset_name: str
     taskflow: BasicTSTaskFlow
     callbacks: List[BasicTSCallback]
@@ -116,6 +120,7 @@ class BasicTSConfig(EasyDict):
 
     ########################### Evaluation Configuration ##########################
 
+    eval_after_train: bool
     save_results: bool # Whether to save evaluation results in a numpy file. Default: False
 
     ############################## Environment Configuration ##############################