| PRTools contents |
DRBMC
W = DRBMC(A)
W = DRBMC(A, N)
W = DRBMC(A, N, L)
| Input | |
| A | Dataset |
| N | Number of hidden units |
| L | Regularization parameter (L2) |
| Output | |
| W | Discriminative Restricted Boltzmann Machine classifier |
The classifier trains a discriminative Restricted Boltzmann Machine (RBM) on dataset A. The discriminative RBM can be viewed as a logistic regressor with hidden units. The discriminative RBM has N hidden units (default = 50). It is trained with L2 regularization using regularization parameter L (default = 0).
New objects are classified in the same way as in a logistic regressor, using a softmax function over the labels.
H. Larochelle and Y. Bengio. Classification using Discriminative Restricted Boltzmann Machines. Proceedings of the 25th International Conference on Machine Learning (ICML), pages 536?543, 2008.
| PRTools contents |