使用PHM2012大赛的轴承数据库,研究如何使用深度学习算法对滚珠轴承进行剩余寿命预测的试错项目
to prediction the remain useful life of bearing based on 2012 PHM data
碎碎念:这个项目一开始只是一个默默无名的研究生用来记录自己在深度学习算法上的学习过程,所以本质上就是一个没啥卵用的项目。结果项目停止更新的两年多总有人会给我发邮件问这个项目,(也有可能相关方向的代码是真的少,所以找到这里。。。),而且我也不做相关方向了,所以做一个总结。
dataset.py
,这是将PHM2012、德国帕德博恩大学的数据库以及cwru数据进行封装,从而让我方便进行数据库替换来验证模型效果。首先要先从网上下载数据库,然后使用dataset.py
里面的make_xxx_dataset()
方法生成pkl文件(有些数据量特别大的还是分成了多个pkl文件),加载的时候使用DataSet.load_dataset()
方法就行了,最后还可以更具条件加载不同的数据,具体看dataset.py
的代码注释。(记得看文件(夹)的调用路径是否需要更改)attention.py
和attention2.py
,具体参考论文A novel deep learning method based on attention mechanism for bearing remaining useful life prediction。dataset.py
. This code is used to package the dataset from PHM2012, the paderborn, the CWRU and the IMS. First, download the dataset in the network, and use the make_xxx_dataset()
function in dataset.py
to generate some pkl files. Then such files are able to load with funciton DataSet.load_dataset()
, specially select the data according to different condition. More detail please read the code comment.attention.py
and attention2.py
. More detail can be found in this article: A novel deep learning method based on attention mechanism for bearing remaining useful life prediction。