Research Article

Convolutional Recurrent Neural Network for Fault Diagnosis of High-Speed Train Bogie

Table 4

Hyperparameters used for comparison experiments.
(a) The architectures based on CNN, RNN, and CRNN

HST SDS200
CRNN1D-CNNLSTM

Conv_1d cells116/232 116/232 -
pool_1d size 3 3 -
RNN cells450/450/450/450- 450/450/450/450
1D-CNN layers 2 2 -
RNN layers 4 - 4
Fully-connected layers 1 1 1
learning rate 0.0050.0003 0.001

(b) The architectures based on ensemble learning

RF XGboost GBDT

estimators size 1000 1000 1000
learning rate - 0.15 0.01
max depth 500 3 8
max features sqrt - sqrt