cwru

Case Western Reserve University Bearing Data


Keywords
cwru bearing fault diagnosis dataset
License
MIT
Install
pip install cwru==0.2

Documentation

Case Western Reserve University Bearing Data

Usage

import cwru
data = cwru.CWRU("12DriveEndFault", "1797", 384)

Then, you can use data.X_train, data.y_train, data.X_test, data.y_test, data.labels, data.nclasses to train and evaluate your data driven fault diagnostic algrithms.

Not have a local copy of CWRU Bearing data on your machine? Don't be worry, These .mat files required by your code will be automatically downloaded from Case Western Reserve University Bearing Data Center, renamed properly with their corresponding class, stored in ~/Datasets/CWRU directory to avoid downloading in future.

Arguments

The constructor of class cwru.CWRU has three arguments:

  • exp: experiment, supporting "12DriveEndFault", "12FanEndFault", "48DriveEndFault"
  • rpm: rpm during testing
  • length: length of the signal slice, namely X_train.shape[1]

Install

Install by pip:

$ pip install --user cwru

Or install manually:

$ python setup.py install

Test

You can run unit test by:

$ python -m unittest discover