Upload README.md with huggingface_hub

README.md

@@ -4,7 +4,7 @@ tags:
   - sklearn
   - predictive-maintenance
   - classification
-  - decision-tree
+  - adaboost
 datasets:
   - jskswamy/predictive-maintenance-data
 metrics:
@@ -17,7 +17,7 @@ metrics:
 
 ## Model Description
 
-This Decision Tree classifier predicts whether a diesel engine requires maintenance based on sensor readings.
+This AdaBoost classifier predicts whether a diesel engine requires maintenance based on sensor readings.
 The model was trained for **commercial fleet predictive maintenance** applications.
 
 ## Primary Metric: F2 Score
@@ -35,7 +35,7 @@ The model is optimized for **F2 Score** (recall weighted 2x over precision) beca
 | Precision | 0.6304 |
 | F1 Score | 0.7733 |
 | Accuracy | 0.6304 |
-| ROC-AUC | 0.6255 |
+| ROC-AUC | 0.6443 |
 
 ## Features
 
@@ -81,7 +81,7 @@ print(f"Maintenance Probability: {probability[0]:.2%}")
 
 ## Training Details
 
-- **Algorithm:** Decision Tree
+- **Algorithm:** AdaBoost
 - **Hyperparameter Tuning:** GridSearchCV with 5-fold stratified CV
 - **Scoring:** F2 Score (beta=2)
 - **Training Data:** 15,628 samples