I want to use BernoulliRBM implementation in scikit-learn for Restricted Boltzmann Machines, but I can’t find anywhere a way or a parameter to set the number of Gibbs steps k for the PCD sampling. Should I assume that k=1 and can't be modified?

有帮助吗?

解决方案

Yes, the training algorithm uses a hardwired "k". It can be seen in the _fit method, which samples once, then updates the parameters.

许可以下: CC-BY-SA归因
不隶属于 StackOverflow
scroll top