/LSTM_fault_detection

Primary LanguagePython

LSTM_fault_detection

  • The code for the paper "Sequential Fault Diagnosis based on LSTM Neural Network"
  • The code is developed under the python package: PyTorch

Requirements for running the python codes "nca.py" and "autoencoder.py"

Please install the following python libraries before running the code
  • python==3.52
  • numpy==1.13.3
  • PyTorch==0.20
  • scikit-learn==0.19.0

For the GPU acceleration with PyTorch, please refer to PyTorch

Run

On linux:

  • python3 nca.py
  • python3 autoencoder.py