Correct values for enc_in, dec_in and c_out

The values did not match the number of features (inclusive OT and exclusive date) of the provided traffic dataset (traffic.csv)
yhy125 · Jan 24, 2023 · 54292bd · 54292bd
1 parent d910070
commit 54292bd
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/scripts/Traffic_script/Reformer.sh b/scripts/Traffic_script/Reformer.sh
@@ -14,9 +14,9 @@ python -u run.py \
   --e_layers 2 \
   --d_layers 1 \
   --factor 3 \
-  --enc_in 1 \
-  --dec_in 1 \
-  --c_out 1 \
+  --enc_in 862 \
+  --dec_in 862 \
+  --c_out 862 \
   --des 'Exp' \
   --itr 1 \
   --train_epochs 3
@@ -35,9 +35,9 @@ python -u run.py \
   --e_layers 2 \
   --d_layers 1 \
   --factor 3 \
-  --enc_in 1 \
-  --dec_in 1 \
-  --c_out 1 \
+  --enc_in 862 \
+  --dec_in 862 \
+  --c_out 862 \
   --des 'Exp' \
   --itr 1 \
   --train_epochs 3
@@ -56,9 +56,9 @@ python -u run.py \
   --e_layers 2 \
   --d_layers 1 \
   --factor 3 \
-  --enc_in 1 \
-  --dec_in 1 \
-  --c_out 1 \
+  --enc_in 862 \
+  --dec_in 862 \
+  --c_out 862 \
   --des 'Exp' \
   --itr 1 \
   --train_epochs 3
@@ -77,9 +77,9 @@ python -u run.py \
   --e_layers 2 \
   --d_layers 1 \
   --factor 3 \
-  --enc_in 1 \
-  --dec_in 1 \
-  --c_out 1 \
+  --enc_in 862 \
+  --dec_in 862 \
+  --c_out 862 \
   --des 'Exp' \
   --itr 1 \
-  --train_epochs 3
+  --train_epochs 3