indianakhil commited on
Commit
f47a6dd
·
verified ·
1 Parent(s): 3a6e03f

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +51 -51
README.md CHANGED
@@ -1,51 +1,51 @@
1
- ---
2
- language: en
3
- license: mit
4
- tags:
5
- - predictive-maintenance
6
- - binary-classification
7
- - engine-health
8
- - scikit-learn
9
- datasets:
10
- - indianakhil/engine-predictive-maintenance
11
- metrics:
12
- - f1
13
- - accuracy
14
- - roc_auc
15
- ---
16
-
17
- # Engine Predictive Maintenance Model
18
-
19
- ## Model Description
20
- Binary classifier predicting engine health (Normal vs Faulty) from six sensor readings.
21
-
22
- - **Model Type**: AdaBoostClassifier
23
- - **Task**: Binary Classification (0=Normal, 1=Faulty)
24
- - **Training Data**: `indianakhil/engine-predictive-maintenance` (19,535 records)
25
- - **Best Hyperparameters**: `{'learning_rate': 0.5, 'n_estimators': 50}`
26
-
27
- ## Performance (Test Set — 20% holdout)
28
-
29
- | Metric | Score |
30
- |---|---|
31
- | Accuracy | 0.6644 |
32
- | Precision | 0.6787 |
33
- | Recall | 0.8883 |
34
- | **F1-Score** | **0.7695** |
35
- | ROC-AUC | 0.6960 |
36
- | CV F1 (5-fold) | 0.7663 |
37
-
38
- ## Input Features
39
- Engine_RPM, Lub_Oil_Pressure, Fuel_Pressure, Coolant_Pressure,
40
- Lub_Oil_Temperature, Coolant_Temperature
41
-
42
- ## Usage
43
- ```python
44
- from huggingface_hub import hf_hub_download
45
- import joblib, pandas as pd
46
- model = joblib.load(hf_hub_download(
47
- repo_id='indianakhil/engine-predictive-maintenance-model',
48
- filename='best_model.pkl'))
49
- pred = model.predict(X) # 0=Normal, 1=Faulty
50
- prob = model.predict_proba(X)[:, 1] # Fault probability
51
- ```
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ tags:
5
+ - predictive-maintenance
6
+ - binary-classification
7
+ - engine-health
8
+ - scikit-learn
9
+ datasets:
10
+ - indianakhil/engine-predictive-maintenance
11
+ metrics:
12
+ - f1
13
+ - accuracy
14
+ - roc_auc
15
+ ---
16
+
17
+ # Engine Predictive Maintenance Model
18
+
19
+ ## Model Description
20
+ Binary classifier predicting engine health (Normal vs Faulty) from six sensor readings.
21
+
22
+ - **Model Type**: AdaBoostClassifier
23
+ - **Task**: Binary Classification (0=Normal, 1=Faulty)
24
+ - **Training Data**: `indianakhil/engine-predictive-maintenance` (19,535 records)
25
+ - **Best Hyperparameters**: `{'learning_rate': 0.5, 'n_estimators': 50}`
26
+
27
+ ## Performance (Test Set — 20% holdout)
28
+
29
+ | Metric | Score |
30
+ |---|---|
31
+ | Accuracy | 0.6644 |
32
+ | Precision | 0.6787 |
33
+ | Recall | 0.8883 |
34
+ | **F1-Score** | **0.7695** |
35
+ | ROC-AUC | 0.6960 |
36
+ | CV F1 (5-fold) | 0.7663 |
37
+
38
+ ## Input Features
39
+ Engine_RPM, Lub_Oil_Pressure, Fuel_Pressure, Coolant_Pressure,
40
+ Lub_Oil_Temperature, Coolant_Temperature
41
+
42
+ ## Usage
43
+ ```python
44
+ from huggingface_hub import hf_hub_download
45
+ import joblib, pandas as pd
46
+ model = joblib.load(hf_hub_download(
47
+ repo_id='indianakhil/engine-predictive-maintenance-model',
48
+ filename='best_model.pkl'))
49
+ pred = model.predict(X) # 0=Normal, 1=Faulty
50
+ prob = model.predict_proba(X)[:, 1] # Fault probability
51
+ ```