VSMCG

This codebase is a PyTorch implementation of VSMCG, the course project of Bayesian Theory and Computation.

If it is helpful for your work, please⭐

Install

git clone https://github.com/hengyuf/VSMCG.git
cd VSMCG

Usage

Data Generation

See data_generate.py.

Train the variational policy

See VI.py.

Sample

See sampler_newmodel.py.

EM algorithm

See EM.py.

Acknowledgements

This codebase is under maintenance by Ruitao Chen, Hengyu Fu, Rongcheng Shi and Yuxiao Chen.