Bearing Intelligent Fault Diagnosis Based on Wavelet Transform and Convolutional Neural Network

Table 4

Detailed parameters of three classifiers.

Classifier	BP neural network (BP)	Random forest	SVM
Parameter	The number of neurons in the first layer is 100, the number of neurons in the second layer is 50, and the training times are set to 300 rounds	The Gini coefficient is used to select nodes, and the number of decision trees is set to 100	The kernel function is Gauss kernel function, and the penalty coefficient is 0.8