xiaoyinya / FDGRU-PyTorch

新的神经网络？ 基于pytorcjh

Fault diagnosis of rotating machinery based on recurrent neural networks

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

Requirments

Python3

PyTorch >= 0.4.0

torchvision == 0.8.2

numpy

Pillow

scikit-learn

scikit-image

matplotlib

Usage

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

2. Run main.py.

Citation

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}
}