An implementation with GCU-Transformer with PyTorch for remaining useful life prediction on C-MAPSS.
Author: Haoren Guo, National University of Singapore
This work is modified based on https://github.com/jiaxiang-cheng/PyTorch-Transformer-for-RUL-Prediction
Simply modify the parameters in train.sh
run bash train.sh
.
Change MODES='Train' to MODES='test' and change the MODEL_PATH to the model you saved.
python==3.8.8
numpy==1.20.1
pandas==1.2.4
matplotlib==3.3.4
pytorch==1.8.1
This work is inpired by Mo, Y., Wu, Q., Li, X., & Huang, B. (2021). Remaining useful life estimation via transformer encoder enhanced by a gated convolutional unit. Journal of Intelligent Manufacturing, 1-10.