feat: 🎸 update data and prediction visualization scripts

GestaltCogTeam · Sep 13, 2024 · 4035980 · 4035980
1 parent ed9e9dc
commit 4035980
Show file tree

Hide file tree

Showing 4 changed files with 127 additions and 802 deletions.
diff --git a/experiments/evaluate.py b/experiments/evaluate.py
@@ -8,11 +8,11 @@
 
 def parse_args():
     parser = ArgumentParser(description="Evaluate time series forecasting model in BasicTS framework!")
-    parser.add_argument("-cfg", "--config", default="examples/complete_config.py", help="training config")
-    parser.add_argument("-ckpt", "--checkpoint", default="")
-    parser.add_argument("-g", "--gpus", default="0")
+    parser.add_argument("-cfg", "--config", default="baselines/STID/PEMS08_LTSF.py", help="training config") # enter your config file path
+    parser.add_argument("-ckpt", "--checkpoint", default="checkpoints/STID/PEMS08_100_336_336/97d131cadc14bd2b9ffa892d59d55129/STID_best_val_MAE.pt") # enter your own checkpoint file path
+    parser.add_argument("-g", "--gpus", default="5")
     parser.add_argument("-d", "--device_type", default="gpu")
-    parser.add_argument("-b", "--batch_size", default=None)
+    parser.add_argument("-b", "--batch_size", default=None) # use the batch size in the config file
 
     return parser.parse_args()