M-NewtonianVAE

This is an extension of NewtonianVAE.

Prerequisites

Install the necessary packages.

python -m pip install -r requirements.txt

Tips

• Recommend using a virtual environment such as venv
• The installation of matplotlib, opencv, and their associated Qt is likely to be environmentally dependent.
• You may need to install PyTorch according to official site.

Run

Create Data

Example:

./src/create_data.py -c config/sim/reacher2d.json5 --mode save-data --save-dir data/reacher2d

If you want to see what kind of data you are looking for:

./src/create_data.py -c CONFIG_FILE.json5 --mode show-plt

Train

./src/train.py -c CONFIG_FILE.json5

Correlation

./src/correlation.py -c CONFIG_FILE.json5