add new test case

ttm-r2-etth-finetuned-impute/config.json

@@ -0,0 +1,70 @@
+{
+  "_name_or_path": "ibm-granite/granite-timeseries-ttm-r2",
+  "adaptive_patching_levels": 3,
+  "architectures": [
+    "TinyTimeMixerForPrediction"
+  ],
+  "categorical_vocab_size_list": null,
+  "context_length": 512,
+  "d_model": 192,
+  "d_model_scale": 3,
+  "decoder_adaptive_patching_levels": 0,
+  "decoder_d_model": 128,
+  "decoder_d_model_scale": 2,
+  "decoder_mode": "mix_channel",
+  "decoder_num_layers": 2,
+  "decoder_raw_residual": false,
+  "distribution_output": "student_t",
+  "dropout": 0.4,
+  "enable_forecast_channel_mixing": false,
+  "exogenous_channel_indices": [
+    1,
+    2,
+    3,
+    4,
+    5,
+    6
+  ],
+  "expansion_factor": 2,
+  "fcm_context_length": 1,
+  "fcm_gated_attn": true,
+  "fcm_mix_layers": 3,
+  "fcm_prepend_past": true,
+  "fcm_prepend_past_offset": null,
+  "fcm_use_mixer": true,
+  "frequency_token_vocab_size": 5,
+  "gated_attn": true,
+  "head_dropout": 0.4,
+  "init_embed": "pytorch",
+  "init_linear": "pytorch",
+  "init_processing": true,
+  "init_std": 0.02,
+  "loss": "mse",
+  "mode": "common_channel",
+  "model_type": "tinytimemixer",
+  "norm_eps": 1e-05,
+  "norm_mlp": "LayerNorm",
+  "num_input_channels": 7,
+  "num_layers": 2,
+  "num_parallel_samples": 100,
+  "num_patches": 8,
+  "patch_last": true,
+  "patch_length": 64,
+  "patch_stride": 64,
+  "positional_encoding_type": "sincos",
+  "post_init": false,
+  "prediction_channel_indices": [
+    0
+  ],
+  "prediction_filter_length": null,
+  "prediction_length": 96,
+  "resolution_prefix_tuning": false,
+  "scaling": "std",
+  "self_attn": false,
+  "self_attn_heads": 1,
+  "stride_ratio": 1,
+  "torch_dtype": "float32",
+  "transformers_version": "4.46.2",
+  "use_decoder": true,
+  "use_positional_encoding": false
+}

ttm-r2-etth-finetuned-impute/generation_config.json

@@ -0,0 +1,4 @@
+{
+  "_from_model_config": true,
+  "transformers_version": "4.46.2"
+}

ttm-r2-etth-finetuned-impute/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4918ebc47cd468280fa8a9db42a4d7285e206526a8a9d2b2d1c78182472ab9ff
+size 3247008

ttm-r2-etth-finetuned-impute/preprocessor_config.json

@@ -0,0 +1,103 @@
+{
+  "categorical_encoder": null,
+  "conditional_columns": [],
+  "context_length": 512,
+  "control_columns": [
+    "HUFL",
+    "HULL",
+    "MUFL",
+    "MULL",
+    "LUFL",
+    "LULL"
+  ],
+  "encode_categorical": false,
+  "feature_extractor_type": "TimeSeriesPreprocessor",
+  "freq": "1h",
+  "frequency_mapping": {
+    "10min": 4,
+    "15min": 5,
+    "2min": 2,
+    "30min": 6,
+    "5min": 3,
+    "W": 9,
+    "d": 8,
+    "h": 7,
+    "min": 1,
+    "oov": 0
+  },
+  "id_columns": [],
+  "observable_columns": [],
+  "prediction_length": 96,
+  "processor_class": "TimeSeriesPreprocessor",
+  "scaler_dict": {
+    "0": {
+      "copy": true,
+      "feature_names_in_": [
+        "HUFL",
+        "HULL",
+        "MUFL",
+        "MULL",
+        "LUFL",
+        "LULL"
+      ],
+      "mean_": [
+        7.807025544026027,
+        1.9638457706364187,
+        4.854088594068457,
+        0.7027733450419731,
+        2.9906340412969477,
+        0.7704704349555455
+      ],
+      "n_features_in_": 6,
+      "n_samples_seen_": 10452,
+      "scale_": [
+        6.134403361410595,
+        2.1455700413380594,
+        5.908511495039344,
+        1.9702885996497426,
+        1.2502961279520424,
+        0.6677933692498211
+      ],
+      "var_": [
+        37.63090460048561,
+        4.603470802287402,
+        34.910508087012055,
+        3.8820371659097437,
+        1.5632404075718698,
+        0.4459479840140279
+      ],
+      "with_mean": true,
+      "with_std": true
+    }
+  },
+  "scaler_type": "standard",
+  "scaling": true,
+  "scaling_id_columns": [],
+  "static_categorical_columns": [],
+  "target_columns": [
+    "OT"
+  ],
+  "target_scaler_dict": {
+    "0": {
+      "copy": true,
+      "feature_names_in_": [
+        "OT"
+      ],
+      "mean_": [
+        17.292530528345626
+      ],
+      "n_features_in_": 1,
+      "n_samples_seen_": 10452,
+      "scale_": [
+        8.513664476018814
+      ],
+      "var_": [
+        72.4824828102247
+      ],
+      "with_mean": true,
+      "with_std": true
+    }
+  },
+  "time_series_task": "forecasting",
+  "timestamp_column": "date"
+}

ttm-r2-etth-finetuned-impute/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ba92e917cfe922e58d80dbf596abde0e26f2626659249831a09ac48bcae6ade
+size 5240

ttm-r2-etth-finetuned-impute/tsfm_config.json

@@ -0,0 +1,16 @@
+{
+  "exogenous_support": true,
+  "is_finetuned": true,
+  "maximum_context_length": 512,
+  "maximum_prediction_length": 96,
+  "minimum_context_length": 512,
+  "model_class_name": "TinyTimeMixerForPrediction",
+  "model_config_name": "TinyTimeMixerConfig",
+  "model_type": "tinytimemixer",
+  "module_path": "tsfm_public",
+  "inference_handler_module_path": "tsfminference.tsfm_inference_handler",
+  "inference_handler_class_name": "TinyTimeMixerForecastingInferenceHandler",
+  "tuning_handler_module_path": "tsfmfinetuning.tsfm_tuning_handler",
+  "tuning_handler_class_name": "TinyTimeMixerForecastingTuningHandler",
+  "extra_pipeline_arguments": {"impute_method": "linear"}  
+}