Skip to content

Code associated with the work presented at ICLR and ICML 2022 workshops

License

Notifications You must be signed in to change notification settings

mariomorvan/Denoising-Time-Series-Transformer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Don’t Pay Attention to the Noise: Learning Self-supervised Light Curve Representations with a Denoising Time Series Transformer

This is the official repo associated with the above work presented at ICLR 2022 AI for Earth & Space Science and ICML 2022 Machine Learning for Astrophysics workshops.

It contains some utilities to process light curve (through Pytorch Dataset and Transform objects), as well as the Denoising Time Series Transformer implemented as a Pytorch lightning module.

We advise to run the code in a dedicated virtual/conda environment after installing the required dependencies:

pip install -r requirements

TESS data used for experiments can be downloaded by executing the bash script tesscurl_sector_1_lc.sh saved in the data directory (and found on STSCI's archive website)

Finally the DTST can be run with:

main.py --train_path [path_to_training_directory]

See options with python main.py --help

About

Code associated with the work presented at ICLR and ICML 2022 workshops

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages