4 Commits
Author SHA1 Message Date
Kashif Rasul cc7dea9f2f Fixed TFT transform 2021-07-07 11:22:20 +02:00
Dr. Kashif Rasul 0ab82da2cb use glutonts 0.8.0 and remove wandb 2021-07-06 13:16:48 +02:00
aslinagyandalex sliz-nagy 5039dcf6ea TFT fixes (#55)
* fixies for running tft

* fix test set configs

* added AgeFeature back to the pipeline

* fixing AgeFeature

* revert setups changes

Co-authored-by: alex sliz-nagy <alex.sliz.nagy@blackswan.com>
2021-06-08 13:23:22 +02:00
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