test_iTransformer.json

{                  
    "model_name": "ITransformer",
    "use_decoder": true,
    "model_type": "PyTorch",
    "model_params": {
      "forecast_history": 96,
      "forecast_length": 96,
      "embed": "timeF",
      "dropout": 0.1,
      "d_model": 512,
      "use_norm": true,
      "targs": 3
   
   },
   "n_targets":3,
    "dataset_params":
    { "class": "TemporalLoader",
      "temporal_feats": ["month", "day", "day_of_week", "hour"],
       "training_path": "tests/test_data/keag_small.csv",
       "validation_path": "tests/test_data/keag_small.csv",
       "test_path": "tests/test_data/keag_small.csv",
       "batch_size":100,
       "forecast_history":96,
       "forecast_length": 96,
       "train_end": 200,
       "valid_start":180,
       "valid_end": 500,
       "test_start":299,
       "test_end": 500,
       "target_col": ["cfs", "precip", "temp"],
       "relevant_cols": ["cfs", "precip", "temp"],
       "scaler": "StandardScaler", 
       "sort_column":"datetime",
       "interpolate": false,
       "feature_param":
     {
         "datetime_params":{
            "month": "numerical",
            "day": "numerical",
            "day_of_week": "numerical",
            "hour":"numerical"
         }
     }
    },
    "early_stopping":
    {
       "patience":3
    },
    "training_params":
    {
       "criterion":"MSE",
       "optimizer": "Adam",
       "optim_params":
       {
         "lr": 0.002
       },

       "epochs": 1,
       "batch_size":5
    
    },
    "GCS": false,
    
    "wandb": {
       "name": "flood_forecast_circleci",
       "tags": ["dummy_run", "circleci"],
       "project":"repo-flood_forecast"
    },
    "forward_params":{
    },
   "metrics":["MSE"],
   "inference_params":
   {     "num_prediction_samples": 100,
         "datetime_start":"2016-05-30",
          "hours_to_forecast":300, 
          "test_csv_path":"tests/test_data/keag_small.csv",
          "decoder_params":{
            "decoder_function": "simple_decode", 
            "unsqueeze_dim": 1},
          "dataset_params":{
             "file_path": "tests/test_data/keag_small.csv",
             "forecast_history":96,
             "forecast_length":96,
             "relevant_cols": ["cfs", "precip", "temp"],
             "target_col": ["cfs", "precip", "temp"],
             "scaling": "StandardScaler",
             "interpolate_param": false,
             "feature_params":
             {
                 "datetime_params":{
                    "month": "numerical",
                    "day": "numerical",
                    "day_of_week": "numerical",
                    "hour":"numerical"
                 }
             },
             "sort_column":"datetime"
          }
      }
   }