diff --git a/deepod/models/time_series/tcned.py b/deepod/models/time_series/tcned.py index 06e7ac3..0d3689a 100644 --- 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 index f0a8b6d..16a3bf9 100644 --- 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