├── .gitignore ├── Dockerfile ├── Makefile ├── README.md ├── autoformer_multivariate.py ├── environment.yml ├── evaluation.py ├── images ├── nhits-arch.png └── results.png ├── long_horizon_baselines.ipynb ├── nhits_intuition.ipynb ├── nhits_multivariate.py ├── residuals.py ├── rnn_multivariate.py └── src ├── __init__.py ├── data ├── __init__.py ├── datasets │ ├── __init__.py │ ├── ecl.py │ ├── epf.py │ ├── ett.py │ ├── favorita.py │ ├── gefcom2012.py │ ├── gefcom2014.py │ ├── m3.py │ ├── m4.py │ ├── m5.py │ ├── tourism.py │ ├── utils.py │ └── wth.py ├── scalers.py ├── tsdataset.py ├── tsloader.py └── utils.py ├── experiments ├── __init__.py └── utils.py ├── losses ├── __init__.py ├── numpy.py ├── pytorch.py └── utils.py └── models ├── __init__.py ├── components ├── __init__.py ├── autocorrelation.py ├── autoformer.py ├── common.py ├── drnn.py ├── embed.py ├── selfattention.py ├── tcn.py └── transformer.py ├── esrnn ├── __init__.py └── esrnn.py ├── nbeats ├── __init__.py ├── ensemble.py └── nbeats.py ├── nhits ├── __init__.py └── nhits.py └── transformer ├── __init__.py ├── autoformer.py ├── informer.py └── transformer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/Dockerfile -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/README.md -------------------------------------------------------------------------------- /autoformer_multivariate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/autoformer_multivariate.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/evaluation.py -------------------------------------------------------------------------------- /images/nhits-arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/images/nhits-arch.png -------------------------------------------------------------------------------- /images/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/images/results.png -------------------------------------------------------------------------------- /long_horizon_baselines.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/long_horizon_baselines.ipynb -------------------------------------------------------------------------------- /nhits_intuition.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/nhits_intuition.ipynb -------------------------------------------------------------------------------- /nhits_multivariate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/nhits_multivariate.py -------------------------------------------------------------------------------- /residuals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/residuals.py -------------------------------------------------------------------------------- /rnn_multivariate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/rnn_multivariate.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/datasets/ecl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/ecl.py -------------------------------------------------------------------------------- /src/data/datasets/epf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/epf.py -------------------------------------------------------------------------------- /src/data/datasets/ett.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/ett.py -------------------------------------------------------------------------------- /src/data/datasets/favorita.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/favorita.py -------------------------------------------------------------------------------- /src/data/datasets/gefcom2012.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/gefcom2012.py -------------------------------------------------------------------------------- /src/data/datasets/gefcom2014.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/gefcom2014.py -------------------------------------------------------------------------------- /src/data/datasets/m3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/m3.py -------------------------------------------------------------------------------- /src/data/datasets/m4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/m4.py -------------------------------------------------------------------------------- /src/data/datasets/m5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/m5.py -------------------------------------------------------------------------------- /src/data/datasets/tourism.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/tourism.py -------------------------------------------------------------------------------- /src/data/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/utils.py -------------------------------------------------------------------------------- /src/data/datasets/wth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/datasets/wth.py -------------------------------------------------------------------------------- /src/data/scalers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/scalers.py -------------------------------------------------------------------------------- /src/data/tsdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/tsdataset.py -------------------------------------------------------------------------------- /src/data/tsloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/tsloader.py -------------------------------------------------------------------------------- /src/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/data/utils.py -------------------------------------------------------------------------------- /src/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/experiments/utils.py -------------------------------------------------------------------------------- /src/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/losses/numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/losses/numpy.py -------------------------------------------------------------------------------- /src/losses/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/losses/pytorch.py -------------------------------------------------------------------------------- /src/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/losses/utils.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/autocorrelation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/autocorrelation.py -------------------------------------------------------------------------------- /src/models/components/autoformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/autoformer.py -------------------------------------------------------------------------------- /src/models/components/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/common.py -------------------------------------------------------------------------------- /src/models/components/drnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/drnn.py -------------------------------------------------------------------------------- /src/models/components/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/embed.py -------------------------------------------------------------------------------- /src/models/components/selfattention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/selfattention.py -------------------------------------------------------------------------------- /src/models/components/tcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/tcn.py -------------------------------------------------------------------------------- /src/models/components/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/components/transformer.py -------------------------------------------------------------------------------- /src/models/esrnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/esrnn/esrnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/esrnn/esrnn.py -------------------------------------------------------------------------------- /src/models/nbeats/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/nbeats/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/nbeats/ensemble.py -------------------------------------------------------------------------------- /src/models/nbeats/nbeats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/nbeats/nbeats.py -------------------------------------------------------------------------------- /src/models/nhits/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/nhits/nhits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/nhits/nhits.py -------------------------------------------------------------------------------- /src/models/transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/transformer/autoformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/transformer/autoformer.py -------------------------------------------------------------------------------- /src/models/transformer/informer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/transformer/informer.py -------------------------------------------------------------------------------- /src/models/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cchallu/n-hits/HEAD/src/models/transformer/transformer.py --------------------------------------------------------------------------------