Skip to content

Instantly share code, notes, and snippets.

@isaacmg
Created January 16, 2024 04:39
Show Gist options
  • Save isaacmg/4ff12f1532058900229261f1963ffd03 to your computer and use it in GitHub Desktop.
Save isaacmg/4ff12f1532058900229261f1963ffd03 to your computer and use it in GitHub Desktop.
{
"model_name": "Crossformer",
"use_decoder": true,
"model_type": "PyTorch",
"model_params": {
"n_time_series": 4,
"forecast_history":6,
"forecast_length": 4,
"seg_len": 6
},
"n_targets": 4,
"dataset_params":
{ "class": "default",
"training_path": "0101100040B_flow.csv",
"validation_path": "0101100040B_flow.csv",
"test_path": "0101100040B_flow.csv",
"forecast_history":6,
"forecast_length":3,
"train_start": 1,
"train_end": 92000,
"valid_start":92001,
"valid_end": 97001,
"test_start":99001,
"test_end": 108000,
"target_col": ["cfs", "p01m", "tmpf", "dwpf"],
"relevant_cols": ["cfs", "p01m", "tmpf", "dwpf"],
"scaler": "StandardScaler",
"interpolate":{
"method":"back_forward_generic",
"params":{
"relevant_columns":["cfs", "tmpf", "p01m", "dwpf"]}
}
},
"training_params":
{
"criterion":"MSE",
"optimizer": "Adam",
"optim_params":
{
"lr": 0.1,
},
"epochs": 5,
"batch_size":200
},
"GCS": false,
"wandb": {
"name": "Crossformer",
"tags": ["crossformer", "ds_testing_dsanet"],
"project": "repo-flood_forecast"
},
"forward_params":{},
"metrics":["MSE"],
"inference_params":
{ "num_prediction_samples": 20,
"datetime_start":"2021-05-31",
"hours_to_forecast":334,
"test_csv_path":"0101100040B_flow.csv",
"decoder_params":{
"decoder_function": "simple_decode",
"unsqueeze_dim": 1}
,
"dataset_params":{
"file_path": "0101100040B_flow.csv",
"forecast_history": 6,
"forecast_length":4,
"relevant_cols": ["cfs", "p01m", "tmpf", "dwpf"],
"target_col": ["cfs", "p01m", "tmpf", "dwpf"],
"scaling": "StandardScaler",
"interpolate_param":{
"method":"back_forward_generic",
"params":{
"relevant_columns":["cfs", "tmpf", "p01m", "dwpf"]}
}
}
}}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment