HoangTrungNguyen commited on
Commit
185d64f
·
verified ·
1 Parent(s): c34c912

Upload HUGGINGFACE.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. HUGGINGFACE.md +3 -1
HUGGINGFACE.md CHANGED
@@ -107,12 +107,14 @@ subprocess.run([
107
  For deep learning, run each model separately:
108
 
109
  ```python
110
- for model_name in ["lstm", "cnn", "tcn"]:
 
111
  subprocess.run([
112
  sys.executable, SCRIPT_PATH,
113
  "--mode", "paper_buildings",
114
  "--target", "Academic",
115
  "--test-span-days", "7",
 
116
  "--dl-models", model_name,
117
  "--epochs", "20",
118
  "--batch-size", "64",
 
107
  For deep learning, run each model separately:
108
 
109
  ```python
110
+ for model_name in ["lstm", "cnn", "tcn", "timesnet"]:
111
+ extra_args = ["--lookback", "144"] if model_name == "timesnet" else []
112
  subprocess.run([
113
  sys.executable, SCRIPT_PATH,
114
  "--mode", "paper_buildings",
115
  "--target", "Academic",
116
  "--test-span-days", "7",
117
+ *extra_args,
118
  "--dl-models", model_name,
119
  "--epochs", "20",
120
  "--batch-size", "64",