Steps to replicate: 1. install the environment 2. ```python train.py --dataset unitedstates```
Steps to replicate:
python train.py --dataset unitedstates