Divya499/ReliabilityPulse
1
ReliabilityPulse: AI-Driven Failure Forecasting for Industrial Assets
ReliabilityPulse is a high-performance predictive maintenance system for smart manufacturing. Built on the AI4I 2020 dataset, it features a modular ML pipeline and a premium Streamlit dashboard. Using XGBoost and sensor analytics (Temp, Torque, RPM), it predicts failures with high precision, minimizing downtime and optimizing machine maintenance.
๐ Live Demo on Hugging Face Spaces
๐ Project Structure
04_predictive_maintenance/
โโโ data/
โ โโโ raw/ai4i2020.csv # Input Dataset (10,000 records)
โ โโโ processed/features.csv # Engineered features and preprocessed data
โโโ models/
โ โโโ xgboost_model.pkl # Primary Classifier (F1 ~88-95%)
โ โโโ isolation_forest.pkl # Anomaly Baseline model
โ โโโ scaler.pkl # StandardScaler for sensors
โโโ pipeline/
โ โโโ 01_eda.py # Visual Analysis (Distributions, Heatmaps)
โ โโโ 02_feature_engineering.py # Physics-based Feature Engineering
โ โโโ 03_preprocessing.py # Scaling and SMOTE Balancing
โ โโโ 04_model_training.py # GridSearch Tuning for best models
โ โโโ 05_evaluation.py # Performance Reporting and Metrics
โโโ outputs/
โ โโโ confusion_matrix.png # Classification Performance Plot
โ โโโ roc_curve_comparison.png # ROC for Logistic, SVM, XGBoost
โ โโโ feature_importance.png # Key risk drivers bar chart
โ โโโ anomaly_scores.png # Isolation Forest Score Distribution
โโโ app.py # Interactive Streamlit Dashboard
โโโ path_utils.py # Centralized Path Management
โโโ README.md # Project Documentation๐ Getting Started
1. Install Dependencies
pip install pandas numpy scikit-learn xgboost imbalanced-learn matplotlib seaborn joblib streamlit2. Run the Pipeline
To retrain the model and generate metrics:
python pipeline/01_eda.py
python pipeline/02_feature_engineering.py
python pipeline/03_preprocessing.py
python pipeline/04_model_training.py
python pipeline/05_evaluation.py3. Launch the Dashboard
streamlit run app.py๐ Performance Summary (XGBoost)
- F1-Score (Failure): Target range 88โ95% achieved.
- Recall (Failure): Optimized to >90% to prevent missed mechanical failures.
- Top Drivers: Tool wear interaction with Torque and Power usage.
๐ง Maintenance Recommendations (Dashboard)
- Low Risk: Schedule routine inspection in 100 hours.
- Medium Risk: Inspect within 24 hours.
- High/Critical Risk: Immediate manual inspection or stop operations.
Built by [Divyanshi Singh](https://www.linkedin.com/in/divyanshi-singh-/) | [GitHub](https://github.com/Divyanshi018572)
