A deep learning–based system for automated fault detection and classification in rotating machines.
├── LICENSE <- Open-source license if one is chosen.
├── Makefile <- Makefile with convenience commands like `make data` or `make train`.
├── README.md <- The top-level README for developers using this project.
├── data
│ ├── external <- Data from third party sources.
│ ├── interim <- Intermediate data that has been transformed.
│ ├── processed <- The final, canonical data sets for modeling.
│ └── raw <- The original, immutable data dump.
│
├── docs <- A default mkdocs project; see www.mkdocs.org for details.
│
├── models <- Trained and serialized models, model predictions, or model summaries.
│
├── notebooks <- Jupyter notebooks that contains code for all the models we trained.
│ │
│ ├── dual_branch_1DCNN.ipynb
│ ├── ResNet_model.ipynb
│ ├── InceptionTime_model.ipynb
│ ├── MiniROCKET_ridge_waveforms.ipynb
│ ├── MiniROCKET_ridge_FFTs.ipynb
│ ├── MiniROCKET_ridge_waveforms_and_FFTs.ipynb
│ └── Multi_scale_dual_encoder_model.ipynb
│
├── pyproject.toml <- Project configuration file with package metadata for
│ rotor_ai and configuration for tools like black.
│
├── references <- Data dictionaries, manuals, and all other explanatory materials.
│
├── reports <- Generated analysis as HTML, PDF, LaTeX, etc.
│ ├── report.pdf <- Detailed report of the project.
│ └── figures <- Generated graphics and figures to be used in reporting.
│
├── requirements.txt <- The requirements file for reproducing the analysis environment, e.g.
│ generated with `pip freeze > requirements.txt`.
│
├── setup.cfg <- Configuration file for flake8.
│
└── rotor_ai <- Source code for use in this project.
│
├── __init__.py <- Makes rotor_ai a Python module.
│
├── config.py <- Store useful variables and configuration.
│
├── data_collection.py <- Code to collect data from the sensor.
│
└── data_ingestion.py <- Code to ingest the data that saves `X_waveforms` and `y`.
You can view the detailed report of our project here.