Update app.py

app.py

@@ -197,8 +197,8 @@ def select_model(horizon, model_type, max_steps=50):
     elif model_type == 'TimesNet':
         return TimesNet(h=horizon,
                         input_size=horizon * 5,
-                        hidden_size=16,
-                        conv_hidden_size=32,
+                        hidden_size=32,
+                        conv_hidden_size=64,
                         loss=HuberMQLoss(level=[90]),
                         scaler_type='standard',
                         learning_rate=1e-3,
@@ -212,16 +212,16 @@ def select_model(horizon, model_type, max_steps=50):
                     input_size=horizon * 5,
                     loss=HuberMQLoss(level=[90]),
                     scaler_type='standard',
-                    encoder_n_layers=2,
-                    encoder_hidden_size=64,
+                    encoder_n_layers=3,
+                    encoder_hidden_size=256,
                     context_size=10,
-                    decoder_hidden_size=64,
-                    decoder_layers=2,
+                    decoder_hidden_size=256,
+                    decoder_layers=3,
                     max_steps=max_steps)
     elif model_type == 'TFT':
         return TFT(h=horizon,
                    input_size=horizon*5,
-                   hidden_size=16,
+                   hidden_size=96,
                    loss=HuberMQLoss(level=[90]),
                    learning_rate=0.005,
                    scaler_type='standard',