Kashif Rasul 0053eceaf1 Temporal fusion transformer (#40) ...

* initial tft model

* fixes

* fix FeatureProjector

* fixed attention module

* added quantile loss output

* added predict

* added tft_transform from gluonts

* requires tensor_split

* comment out key_padding_mask

give nans on GPU

* added example

2021-04-15 19:10:55 +02:00

..

__init__.py

Temporal fusion transformer (#40)

2021-04-15 19:10:55 +02:00

tft_estimator.py

Temporal fusion transformer (#40)

2021-04-15 19:10:55 +02:00

tft_modules.py

Temporal fusion transformer (#40)

2021-04-15 19:10:55 +02:00

tft_network.py

Temporal fusion transformer (#40)

2021-04-15 19:10:55 +02:00

tft_output.py

Temporal fusion transformer (#40)

2021-04-15 19:10:55 +02:00

tft_transform.py

Temporal fusion transformer (#40)

2021-04-15 19:10:55 +02:00