search

ZhaoZhibin / DL-based-Intelligent-Diagnosis-Benchmark

Source codes for the paper "Deep Learning Algorithms for Rotating Machinery Intelligent Diagnosis: An Open Source Benchmark Study"
MIT License
592 stars 164 forks source link

Possible error in CWRUFFT.py #7

Closed Lucashsmello closed 2 years ago

Lucashsmello commented 2 years ago

Hi. The file CNN_Datasets/R_NA/datasets/CWRUFFT.py, at line 105, adds noise to the training dataset, despite being used for a classification task. Is that correct?

ZhaoZhibin commented 2 years ago

In the paper, we also discussed the data augmentation, and thus RandomAddGaussian() means randomly adding Gaussian noise in each training sample and is one kind of the data augmentation method. Data augmentation is a common step in image classification, which can sometimes enhance the generalization ability of the model. Thus, we discussed it.

ZhaoZhibin commented 2 years ago

Hi. The file CNN_Datasets/R_NA/datasets/CWRUFFT.py, at line 105, adds noise to the training dataset, despite being used for a classification task. Is that correct?

In the paper, we also discussed the data augmentation, and thus RandomAddGaussian() means randomly adding Gaussian noise in each training sample and is one kind of the data augmentation method. Data augmentation is a common step in image classification, which can sometimes enhance the generalization ability of the model. Thus, we discussed it.

Lucashsmello commented 2 years ago

Thanks for your response. That is right. However, shouldn't datasets on R_NA folder be free of data augmentation?

ZhaoZhibin commented 2 years ago

Thanks for your response. That is right. However, shouldn't datasets on R_NA folder be free of data augmentation?

Yes, you are right! We have made a mistake when cleaning up our codes. I have modified this error, which happens in both AE and CNN

Lucashsmello commented 2 years ago

Nice! Many thanks. This seems to happen to other datasets in CNN_Datasets/R_NA folder too:

ZhaoZhibin commented 2 years ago

Nice! Many thanks. This seems to happen to other datasets in CNN_Datasets/R_NA folder too:

  • CWRUCWT.py
  • JNUCWT.py
  • MFPTCWT.py
  • PUCWT.py
  • PU.py
  • SEUCWT.py
  • UoCCWT.py
  • XJTUCWT.py They all use RandomScaler and RandomCrop.

Thanks very much!

Lucashsmello commented 2 years ago

Resolved.