You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
2. Train the model (~2 hours on Tesla K80 GPU with default hyperparams):
13
+
2. Train the model (~1.5 hours on Tesla K80 GPU with default hyperparams):
14
14
-`$ cd src && python lstnet.py --gpus=0`
15
15
16
16
## Results & Comparison
17
17
18
18
- The model in the paper predicts with h = 3 on electricity dataset, achieving *RSE = 0.0906, RAE = 0.0519 and CORR = 0.9195* on test dataset
19
-
- This MXNet implementation achieves *RSE = 0.0967, RAE = 0.0581 and CORR = 0.8941* after 100 epochs on the validation dataset
19
+
- This MXNet implementation achieves *RSE = 0.0880, RAE = 0.0542* after 100 epochs on the validation dataset
20
20
- Saved model checkpoint files can be found in `models/`
21
21
22
-

23
-
24
22
## Hyperparameters
25
23
26
24
The default arguements in `lstnet.py` achieve equivolent performance to the published results. For other datasets, the following hyperparameters provide a good starting point:
0 commit comments