dhani10 commited on
Commit
b833cc7
·
verified ·
1 Parent(s): 3c8ed37

Upload README.md with huggingface_hub

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