# Learning World Model with Identifiable Factorization
Experiment results on synthetic data are showcased in Jupyter Notebooks in `/tests` folder. Each notebook contains the scripts for analysis and visualization for one specific experiment.

Run the scripts in `scripts/run_IFactor.sh` to generate results for experiment.

Further details are documented within the code. 

We have anonymized all of the paths used in the project. To reproduce the results, change the path to your project path. 

### Requirements
To install it, create a conda environment with `Python>=3.7` and follow the instructions below. Note, that the current implementation of LEAP requires a GPU.
```
conda create -n IFactor python=3.8
pip install -e .
```

### Datasets
Synthetic data: `python IFactor/tools/gen_dataset.py `

