ThabangTheActuaryCoder/mining-equipment-failure-model
1
Mining Equipment Failure Prediction Model
A GradientBoostingClassifier pipeline for predicting equipment failures, trained on South African mining data.
Intended Use
This model is intended for educational and demonstration purposes as part of an end-to-end ML pipeline showcasing Databricks, MLflow, Azure ML, and Hugging Face Hub integration.
Model Details
Evaluation Metrics
Confusion Matrix
ROC Curve
Feature Importance
Features
Numeric: temperature_celsius, vibration_mm_s, oil_pressure_kpa, rpm, operating_hours, days_since_maintenance, load_percentage, ambient_temperature_celsius, hydraulic_pressure_kpa, num_previous_failures
Categorical: equipment_type, mine_type, shift, province
Sample Usage
import joblib
from huggingface_hub import hf_hub_download
import pandas as pd
# Download and load the model
model_path = hf_hub_download(
repo_id="ThabangTheActuaryCoder/mining-equipment-failure-model",
filename="equipment_failure_model.joblib",
)
model = joblib.load(model_path)
# Create a sample input
sample = pd.DataFrame([{"temperature_celsius": 0, "vibration_mm_s": 0, "oil_pressure_kpa": 0, "rpm": 0, "operating_hours": 0, "days_since_maintenance": 0, "load_percentage": 0, "ambient_temperature_celsius": 0, "hydraulic_pressure_kpa": 0, "num_previous_failures": 0, "equipment_type": 0, "mine_type": 0, "shift": 0, "province": 0}])
# Predict
prediction = model.predict(sample)
probabilities = model.predict_proba(sample)
print(f"Prediction: {prediction}, Probabilities: {probabilities}")