An implementation of the paper[1] in which the authors proposed a method to track the partition function of an RBM during learning.
To validate the implementation, this library also implements AIS[2] applied to RBMs.
pip install git+https://github.com/Ktakuya332C/tracker
See examples
directory.
- G. Desjardins et al. (2011), On tracking the partition function, NIPS
- R.M. Neal (2001), Annealed importance sampling, Stat. Comput.