Home

Awesome

Deep Demand Forecast Models

Pytorch Implementation of DeepAR, MQ-RNN, Deep Factor Models, LSTNet, and TPA-LSTM. Furthermore, combine all these model to deep demand forecast model API.

Requirements

Please install Pytorch before run it, and

pip install -r requirements.txt

Run tests

# DeepAR
python deepar.py -e 100 -spe 3 -nl 1 -l g -not 168 -sp -rt -es 10 -hs 50  -sl 60 -ms

# MQ-RNN
python mq_rnn.py -e 100 -spe 3 -nl 1 -sp -sl 72 -not 168 -rt -ehs 50 -dhs 20 -ss -es 10 -ms

# Deep Factors
python deep_factors.py -e 100 -spe 3 -rt -not 168 -sp -sl 168 -ms

# TPA-LSTM
python tpa_lstm.py -e 1000 -spe 1 -nl 1 -not 168 -sl 30 -sp -rt -max

DeepAR
alt text
MQ-RNN
alt text
Deep Factors
alt text
TPA-LSTM
alt text

Arguments

ArgumentsDetails
-enumber of episodes
-spesteps per episode
-slsequence length
-notnumber of observations to train
-msmean scaler on y
-maxmax scaler on y
-nlnumber of layers
-llikelihood to select, "g" or "nb"
-rtrun test data
-sample_sizesample size to sample after </br> training in deep factors/deepar, default 100

TO DO

Demand Forecast Dataset Resources

Reference