maxjcohen / transformer

Implementation of Transformer model (originally from Attention is All You Need) applied to Time Series.
https://timeseriestransformer.readthedocs.io/en/latest/
GNU General Public License v3.0
842 stars 165 forks source link

A more detailed Readme. #19

Closed tcapelle closed 3 years ago

tcapelle commented 4 years ago

Hello, I am also interested on Transformers applied to TimeSeries, but I am having a hard time understanding the repo. I would suggest adding more info to the Readme to show the capabilities of the library.

I would also be nice to have a small graph like the ones on the training notebooks on the Readme, to show the potential. Great work btw, and I am very interested to collaborate. We run a Time Series Study group on the fast.ai forums:

maxjcohen commented 4 years ago

Hi, the Transformer implemented in this repo is very similar to the original model described in Attention Is All You Need, I would suggest heading there for more information. To answer your questions,

If you want to add some modules, or modify the model itself (for forcasting or classification for example), don't hesitate to fork and PR. And thanks for the links, I'll be sure to check them out !