YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Predictive Maintenance Model: Gradient Boosting
This repository hosts the trained machine learning model for predicting engine condition as part of the Predictive Maintenance project.
Model Details
- Model Type: Gradient Boosting
- Objective: Binary Classification (Normal/Faulty Engine)
- Training Data: Derived from the PredictiveMaintenanceDataset on Hugging Face Hub (sammysri12/PredictiveMaintenanceDataset)
- Preprocessing: Data cleaning (outlier capping) and feature engineering were applied.
- Tuning Metric: F1-Score (due to class imbalance)
Evaluation Metrics (on Test Set)
- F1-Score: 0.7766
- Accuracy: 0.6493
- Precision: 0.6490
- Recall: 0.9667
- ROC-AUC: 0.6894
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support