predictiveanalysis / README.md
amitcoolll's picture
Upload README.md with huggingface_hub
e5e6227 verified

Predictive Maintenance – Engine Condition Model

This repository contains the best-performing model trained on engine sensor data to predict the target: Engine_Condition.

βœ… Best Model

  • Algorithm: GradientBoosting
  • Test F1-score: 0.7601
  • Test Accuracy: 0.6657
  • Best Parameters: {"clf__learning_rate": 0.1, "clf__max_depth": 3, "clf__n_estimators": 100}

πŸ“Š Features Used

  • Engine_RPM
  • Lub_Oil_Pressure
  • Fuel_Pressure
  • Coolant_Pressure
  • Lub_Oil_Temperature
  • Coolant_Temperature

πŸ”§ Training Setup

  • Models tried: DecisionTree, RandomForest, GradientBoosting
  • Cross-validation: 3-fold
  • Scoring metric: F1
  • Preprocessing: Median imputation + StandardScaler