Currently, the reimplementation of the DeepAR paper(DeepAR: Probabilistic Forecasting with Autoregressive Recurrent Networks https://arxiv.org/abs/1704.04110) is available in PyTorch. More papers will be coming soon.
-
Yunkai Zhang([email protected]) - University of California, Santa Barbara
-
Qiao Jiang - Brown University
-
Xueying Ma - Columbia University
-
Acknowledgement: Professor Xifeng Yan's group at UC Santa Barbara. Part of the work was done at WeWork.
-
Install all dependencies listed in requirements.txt. Note that the model has only been tested in the versions shown in the text file.
-
Download the dataset and preprocess the data:
python preprocess_elect.py
-
Start training:
python train.py
-
If you want to perform ancestral sampling,
python train.py --sampling
-
If you do not want to do normalization during evaluation,
python train.py --relative-metrics
-
-
Evaluate a set of saved model weights:
python evaluate.py
-
Perform hyperparameter search:
python search_params.py
The model is evaluated on the electricity dataset, which contains the electricity consumption of 370 households from 2011 to 2014. Under hourly frequency, we use the first week of September, 2014 as the test set and all time steps prior to that as the train set. Following the experiment design in DeepAR, the window size is chosen to be 192, where the last 24 is the forecasting horizon. History (number of time steps since the beginning of each household), month of the year, day of the week, and hour of the day are used as time covariates. Notice that some households started at different times, so we only use windows that contain non-missing values.
Under Gaussian likelihood, we use the Adam optimizer with early stopping to train the model for 20 epoches. The same set of hyperparameters is used as outlined in the paper. Weights with the best ND value is selected, where ND = 0.06349, RMSE = 0.452, rou90 = 0.034 and rou50 = 0.063.
Sample results on electricity. The top 10 plots are sampled from the test set with the highest 10% ND values, whereas the bottom 10 plots are sampled from the rest of the test set.
