Table 3.
Hyperparameter settings of restricted Boltzmann machine-capsule networks (RBM-CapsNet).
| Hyperparameter | Setting |
|---|---|
| Numbers of RBM | 2 |
| Number of nodes in the hidden feature layer | 256 |
| Number of nodes in the PrimaryCaps layer | 128 |
| Iteration of RBM | 100 |
| Iteration of network | 200 |
| Learning rate of RBM | 0.001 |
| Learning rate of network | 0.005 |
| Activation | Relu |
| Batch size | 148 |
| Optimizer | Adam |
| Routing time | 2 |
| Dimension of each capsule | 8 |
| Length of PrimaryCaps | 2 |
| Length of DigitCaps | 2 |