TABLE 5:
The architecture of rich model in PTBDB, which includes a total of 688.8k parameters.
| Layer | Type | Hyper-parameters | Activation |
|---|---|---|---|
| 1 | Split | n_seg=10 | |
| 2 | Convolution1D | n_filter=128, kernel_size=16, stride=2 | ReLU |
| 3 | Convolution1D | n_filter=128, kernel_size=16, stride=2 | ReLU |
| 4 | Convolution1D | n_filter=128, kernel_size=16, stride=2 | ReLU |
| 5 | AveragePooling1D | ||
| 6 | LSTM | hidden_units=128 | ReLU |
| 7 | PositionAttention | ||
| 8 | Dense | hidden_units=n_classes | Linear |
| 9 | Softmax |