Skip to content

Latest commit

 

History

History
50 lines (35 loc) · 2.17 KB

README.md

File metadata and controls

50 lines (35 loc) · 2.17 KB

Notebooks to generate figures

Probabilistic distribution figures

Generate models

$ python pred/train.py data/azure/2019/scaled_by_day8_max40_sample8_day8-12.pkl --tool=darts --context-len=60 --pred-len=40 --lr=1e-4 --epochs=200 --blocks=2 --stacks=3 --layers=4 --model-name=nhits --layer-width=256 --batch-size=32 --dropout=0.1
$ python pred/train.py data/azure/2019/scaled_by_day8_max40_sample8_day8-12.pkl --tool=darts --context-len=60 --pred-len=40 --lr=1e-4 --epochs=200 --blocks=2 --stacks=3 --layers=4 --model-name=nhits --layer-width=256 --batch-size=32 --dropout=0.1 --likelihood=gaussian

This will output trained models at results/pred/scaled_by_day8_max40_sample8_day8-12.

Generate plots

Run plot_prob_pred.ipynb

Utility timeline and other figures

See plot_utility_timeline.ipynb, plot_stats.ipynb, plot_stats_mixed.ipynb

Breakdown figures

See plot_breakdown.ipynb

Solver figures

See plot_solver.ipynb