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
-
Modify the path and model parameters in the main.py.
-
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}
}