Skip to main content
. 2020 Jan 28;10:1631. doi: 10.3389/fphar.2019.01631

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