debasishdas1985 commited on
Commit
13fcdab
·
verified ·
1 Parent(s): d93abea

Add model card

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - tabular-classification
5
+ - predictive-maintenance
6
+ - xgboost
7
+ library_name: scikit-learn
8
+ ---
9
+
10
+ # Engine Predictive Maintenance — XGBoost Classifier
11
+
12
+ Binary classifier predicting whether an engine requires maintenance (`engine_condition = 1`)
13
+ based on six sensor readings (RPM, lub-oil pressure/temp, fuel pressure,
14
+ coolant pressure/temp).
15
+
16
+ ## Usage
17
+
18
+ ```python
19
+ import joblib, pandas as pd
20
+ from huggingface_hub import hf_hub_download
21
+
22
+ path = hf_hub_download(repo_id="debasishdas1985/engine-predictive-maintenance-model", filename="best_engine_model.joblib")
23
+ model = joblib.load(path)
24
+ sample = pd.DataFrame([{
25
+ "engine_rpm": 700, "lub_oil_pressure": 2.5, "fuel_pressure": 11.8,
26
+ "coolant_pressure": 3.2, "lub_oil_temp": 84.1, "coolant_temp": 81.6,
27
+ }])
28
+ print(model.predict(sample), model.predict_proba(sample))
29
+ ```
30
+
31
+ ## Best hyperparameters
32
+ {'colsample_bytree': 1.0, 'learning_rate': 0.1, 'max_depth': 7, 'n_estimators': 200, 'subsample': 0.8}
33
+
34
+ ## Test metrics
35
+ | metric | value |
36
+ |---|---|
37
+ | accuracy | 0.6330 |
38
+ | precision | 0.7323 |
39
+ | recall | 0.6585 |
40
+ | f1-score | 0.6935 |
41
+ | ROC-AUC | 0.6792 |