This is the implementation of the paper “Fault diagnosis of rotating machinery based on recurrent neural networks” in PyTorch.

Python3

PyTorch >= 0.4.0

torchvision == 0.8.2

numpy

Pillow

scikit-learn

scikit-image

matplotlib

1. Modify the path and model parameters in the main.py.

2. Run main.py.

If you found this code useful, please consider citing our paper

@article{zhang2021fault,
  title={Fault diagnosis of rotating machinery based on recurrent neural networks},
  author={Yahui Zhang, Taotao Zhou, Xufeng Huang, Longchao Cao, Qi Zhou},
  journal={Measurement},
  year={2021}
}