State Detection with SOM-VAE

PyTorch Lightning implementation of SOM-VAE by Fortuin et al. (ICLR 2019): https://arxiv.org/abs/1806.02199

• to generate data, run scripts/generate_data.py
• to train SOM-VAE, run scripts/train_model.py
• to visualize the predicted states, run scripts/create_figures.py