Please add the citation to Rupesh Srivastava et al. like [here](https://github.com/lisa-lab/pylearn2/blob/master/pylearn2/models/rbm.py#L133) in LWTA class. 

You should also write the parameters section as it is in BlockGibbsSampler class. (Following [numpy docstring standards](https://github.com/numpy/numpy/blob/master/doc/HOWTO_DOCUMENT.rst.txt#sections)).

Lines 14-16 should be moved to the end of the docstring in a Notes section.

All this docstring formatting helps to automatically generate [documentation](http://deeplearning.net/software/pylearn2/library/models.html).  

Thank you!