Skip to content

Commit 84c6020

Browse files
committed
update configs.yaml
1 parent 7561291 commit 84c6020

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

deepod/models/time_series/tcned.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99

1010

1111
class TcnED(BaseDeepAD):
12-
def __init__(self, seq_len=100, stride=1, epochs=10, batch_size=32, lr=1e-3,
13-
rep_dim=32, hidden_dims=32, kernel_size=3, act='ReLU', bias=False, dropout=0.2,
12+
def __init__(self, seq_len=100, stride=1, epochs=10, batch_size=32, lr=1e-4,
13+
rep_dim=32, hidden_dims=32, kernel_size=3, act='LeakyReLU', bias=False, dropout=0.2,
1414
epoch_steps=-1, prt_steps=10, device='cuda',
1515
verbose=2, random_state=42):
1616
super(TcnED, self).__init__(

testbed/configs.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,12 @@ COUTA:
3535
epochs: 20
3636
batch_size: 64
3737

38+
TcnED:
39+
epochs: 10
40+
lr: 0.0001
41+
rep_dim: 64
42+
hidden_dims: 32
43+
3844
AnomalyTransformer:
3945
lr: 0.0001
4046
epochs: 10
@@ -53,6 +59,8 @@ TimesNet:
5359
top_k: 5
5460
num_kernels: 6
5561

62+
63+
5664
RoSAS:
5765
epoch_steps: 16
5866
epochs: 200

0 commit comments

Comments
 (0)