netket.models.RBM¶
-
class
netket.models.
RBM
(dtype=<class 'numpy.float64'>, activation=<function log_cosh>, alpha=1, use_hidden_bias=True, use_visible_bias=True, precision=None, kernel_init=<function normal.<locals>.init>, hidden_bias_init=<function normal.<locals>.init>, visible_bias_init=<function normal.<locals>.init>, parent=<flax.linen.module._Sentinel object>, name=None)[source]¶ Bases:
flax.linen.module.Module
A restricted boltzman Machine, equivalent to a 2-layer FFNN with a nonlinear activation function in between.
- Attributes
-
if True uses a bias in the dense layer (hidden layer bias).