MPVAE-Pytorch

The re-implementation of MPVAE using PyTorch.

Sample Dataset

We use emotions as our running example since it is commonly used and has a moderate size. Dataset location: datasets/emotions

Dependencies

• Python 3.7+
• PyTorch 1.8.1
• numpy 1.17.3
• sklearn 0.22.1

Older versions might work as well.

Run

python3 run.py

The seed is 8 by default, but can be changed in the config.py file.