update configs.yaml

xuhongzuo · Sep 6, 2023 · 84c6020 · 84c6020
1 parent 7561291
commit 84c6020
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/deepod/models/time_series/tcned.py b/deepod/models/time_series/tcned.py
@@ -9,8 +9,8 @@
 
 
 class TcnED(BaseDeepAD):
-    def __init__(self, seq_len=100, stride=1, epochs=10, batch_size=32, lr=1e-3,
-                 rep_dim=32, hidden_dims=32, kernel_size=3, act='ReLU', bias=False, dropout=0.2,
+    def __init__(self, seq_len=100, stride=1, epochs=10, batch_size=32, lr=1e-4,
+                 rep_dim=32, hidden_dims=32, kernel_size=3, act='LeakyReLU', bias=False, dropout=0.2,
                  epoch_steps=-1, prt_steps=10, device='cuda',
                  verbose=2, random_state=42):
         super(TcnED, self).__init__(

diff --git a/testbed/configs.yaml b/testbed/configs.yaml
@@ -35,6 +35,12 @@ COUTA:
   epochs: 20
   batch_size: 64
 
+TcnED:
+  epochs: 10
+  lr: 0.0001
+  rep_dim: 64
+  hidden_dims: 32
+
 AnomalyTransformer:
   lr: 0.0001
   epochs: 10
@@ -53,6 +59,8 @@ TimesNet:
   top_k: 5
   num_kernels: 6
 
+
+
 RoSAS:
   epoch_steps: 16
   epochs: 200