Skip to main content
. 2021 Apr 4;21(7):2524. doi: 10.3390/s21072524
Algorithm 3 Bearing fault diagnosis method based on Feature Fusion
  • Input: 

    bearing signal sequence f(t),t=1,2,,mλ, window width λ, wavelet packet scale l, wavelet packet function type, Number of runs τ.

  • Output: 

    classification accuracy η, calculation time, Variance of classification accuracy δ.

  • 1:

    The feature matrix of bearing fault is obtained Tm×r by Algorithm 2;

  • 2:

    The feature matrix Tm×r is randomly divided into the training set and test set, and different state types are labeled;

  • 3:

    The SVM classifier is trained by the training set to obtain the SVM-based classification model;

  • 4:

    The test set is input to the SVM-based classification model to obtain the predicted label of the test set. The actual label and predicted label of the test set are calculated according to Equation (26) to calculate the classification accuracy η of the diagnostic model. The total running time of bearing signal from MVSVD feature extraction to training SVM classification model to test result is calculated;

  • 5:

    The model is run τ times in sequence to get the classification accuracy η each time, and the variance δ of classification accuracy is obtained according to Equation (27).