• Quick installation of conda environment for Python: conda env create -f environment.yml

To use DeepHAM to solve the competitive equilibrium of the KS model, run

python train_KS.py

To evaluate the Bellman error of the solution of the KS model, run

python validate_KS.py

Sample scripts for solving the KS model in the Slurm system are provided in the folder src/slurm_scripts

python train_JFV.py

python validate_JFV.py

Details on the model setup and algorithm can be found in our paper.

If you find this work helpful, please consider starring this repo and citing our paper using the following Bibtex.

@article{HanYangE2021deepham,
  title={Deep{HAM}: A global solution method for heterogeneous agent models with aggregate shocks},
  author={Han, Jiequn and Yang, Yucheng and E, Weinan},
  journal={arXiv preprint arXiv:2112.14377},
  year={2021}
}

Please contact us at [email protected] and [email protected] if you have any questions.