SharleyK/engine-predictive-maintenance-model

Engine Predictive Maintenance Model

This repository hosts a Random Forest Classifier model for predicting engine condition (Normal/Faulty) based on sensor readings.

Model Description

The model is a RandomForestClassifier trained on engine sensor data to identify potential failures. It was selected as the best-performing model due to its high recall score, which is critical for minimizing false negatives in predictive maintenance scenarios.

Training Details

• Model Type: Random Forest Classifier
• Objective: Binary Classification (0: Normal, 1: Faulty)
• Training Data: Preprocessed engine sensor data (features: engine_rpm, lub_oil_pressure, fuel_pressure, coolant_pressure, lub_oil_temp, coolant_temp, pressure_ratio, temp_diff; target: engine_condition).
• Key Hyperparameters (tuned):
  • n_estimators: 50
  • max_depth: 5
  • min_samples_split: 5
  • min_samples_leaf: 1

Evaluation Metrics (on Test Set)

• Accuracy: 0.6601
• Precision: 0.6652
• Recall: 0.9277 (Prioritized metric)
• F1-Score: 0.7748

How to Use

To load and use this model for inference:

from huggingface_hub import hf_hub_download
import joblib
import pandas as pd

model_path = hf_hub_download(repo_id="SharleyK/engine-predictive-maintenance-model", filename="random_forest_model.pkl")
model = joblib.load(model_path)

# Example inference (replace with your actual data)
# Make sure your input data has the same features and preprocessing steps as training
example_data = pd.DataFrame([{
    'engine_rpm': 700.0, 'lub_oil_pressure': 2.5, 'fuel_pressure': 11.8, 'coolant_pressure': 3.2,
    'lub_oil_temp': 80.0, 'coolant_temp': 82.0, 'pressure_ratio': 0.78, 'temp_diff': -2.0
}])

prediction = model.predict(example_data)
print(f"Predicted Engine Condition: {prediction[0]} (0=Normal, 1=Faulty)")

License

[Specify license, e.g., MIT, Apache 2.0]