spac1ngcat commited on
Commit
a558c12
·
verified ·
1 Parent(s): 2b94bce

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. best_model.joblib +3 -0
  2. model_metadata.json +46 -0
best_model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4606691dfb67992e0f1f1de65ea188637c139a878c39586b277e1b39d8f200b4
3
+ size 247400
model_metadata.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "Gradient Boosting Classifier",
3
+ "best_params": {
4
+ "ccp_alpha": 0.0,
5
+ "criterion": "friedman_mse",
6
+ "init": null,
7
+ "learning_rate": 0.015327875046521984,
8
+ "loss": "log_loss",
9
+ "max_depth": 5,
10
+ "max_features": null,
11
+ "max_leaf_nodes": null,
12
+ "min_impurity_decrease": 0.0,
13
+ "min_samples_leaf": 10,
14
+ "min_samples_split": 20,
15
+ "min_weight_fraction_leaf": 0.0,
16
+ "n_estimators": 50,
17
+ "n_iter_no_change": null,
18
+ "random_state": 42,
19
+ "subsample": 1.0,
20
+ "tol": 0.0001,
21
+ "validation_fraction": 0.1,
22
+ "verbose": 0,
23
+ "warm_start": false
24
+ },
25
+ "test_metrics": {
26
+ "accuracy": 0.6512540522095206,
27
+ "precision": 0.6502274795268426,
28
+ "recall": 0.9669824086603518,
29
+ "f1_score": 0.7775843307943416,
30
+ "roc_auc": 0.694657063212122,
31
+ "pr_auc": 0.7943643677015135
32
+ },
33
+ "feature_columns": [
34
+ "Engine_RPM",
35
+ "Lub_Oil_Pressure",
36
+ "Fuel_Pressure",
37
+ "Coolant_Pressure",
38
+ "Lub_Oil_Temperature",
39
+ "Coolant_Temperature"
40
+ ],
41
+ "target_column": "Engine_Condition",
42
+ "class_mapping": {
43
+ "0": "Normal",
44
+ "1": "Faulty"
45
+ }
46
+ }