It would be nice to define some easier imports for the submodules. For instance, instead of doing from rbms.bernoulli_bernoulli.functional import sample_hiddens, do from rbms.bernoulli_bernoulli import sample_hiddens.
I think it would be nice to have BBRBM, PBRBM and other classes imported directly as from rbms import BBRBM
It would be nice to define some easier imports for the submodules. For instance, instead of doing
from rbms.bernoulli_bernoulli.functional import sample_hiddens, dofrom rbms.bernoulli_bernoulli import sample_hiddens.I think it would be nice to have
BBRBM,PBRBMand other classes imported directly asfrom rbms import BBRBM