File size: 650 Bytes
b833cc7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
---
tags:
- sklearn
- predictive-maintenance
- classification
- automotive
library_name: sklearn
license: apache-2.0
pipeline_tag: text-classification
---
# Engine Predictive Maintenance Model
**Best model:** `GradientBoosting`
**F1 (test):** 0.7657
## Usage
```python
import pickle, pandas as pd
with open('best_model.pkl', 'rb') as f:
model = pickle.load(f)
# X = pd.DataFrame([{
# 'engine_rpm': 2500, 'lub_oil_pressure': 4.2, 'fuel_pressure': 110,
# 'coolant_pressure': 1.2, 'lub_oil_temp': 95, 'coolant_temp': 88
# }])
# pred = model.predict(X)[0]
```
## Notes
- Tuned via GridSearchCV (scoring=F1), metrics logged with MLflow. |