凯斯西储大学轴承数据
cwru的Python项目详细描述
用法
importcwrudata=cwru.CWRU("12DriveEndFault","1797",384)
然后,可以使用data.X_train,data.y_train,data.X_test,data.y_test,data.labels,data.nclasses来训练和评估数据驱动的故障诊断算法
您的机器上没有CWRU轴承数据的本地副本?不用担心,这些代码所需的.mat文件将自动从Case Western Reserve University Bearing Data Center下载,并使用相应的类正确重命名,存储在~/Datasets/CWRU目录中,以避免以后下载
参数
类cwru.CWRU的构造函数有三个参数:
- exp:实验,支持"12DriveEndFault", "12FanEndFault", "48DriveEndFault"
- rpm:测试期间的rpm
- 长度:信号片的长度,即X_train.shape[1]
安装
按PIP安装:
$ pip install –user cwru
或手动安装:
$ python setup.py install
测试
可以通过以下方式运行单元测试:
$ python -m unittest discover