Bernoulli Restricted Boltzmann Machine (RBM) Parameters ---------- n_hidden : int, default 128 The number of hidden units. learning_rate : float, default 0.1 batch_size : int, default 10 max_epochs : int, default 100
(self, n_hidden=128, learning_rate=0.1, batch_size=10, max_epochs=100)
| 20 | y_required = False |
| 21 | |
| 22 | def __init__(self, n_hidden=128, learning_rate=0.1, batch_size=10, max_epochs=100): |
| 23 | """Bernoulli Restricted Boltzmann Machine (RBM) |
| 24 | |
| 25 | Parameters |
| 26 | ---------- |
| 27 | |
| 28 | n_hidden : int, default 128 |
| 29 | The number of hidden units. |
| 30 | learning_rate : float, default 0.1 |
| 31 | batch_size : int, default 10 |
| 32 | max_epochs : int, default 100 |
| 33 | """ |
| 34 | self.max_epochs = max_epochs |
| 35 | self.batch_size = batch_size |
| 36 | self.lr = learning_rate |
| 37 | self.n_hidden = n_hidden |
| 38 | |
| 39 | def fit(self, X, y=None): |
| 40 | self.n_visible = X.shape[1] |
nothing calls this directly
no outgoing calls
no test coverage detected