File size: 1,114 Bytes
a558c12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
{
  "model_type": "Gradient Boosting Classifier",
  "best_params": {
    "ccp_alpha": 0.0,
    "criterion": "friedman_mse",
    "init": null,
    "learning_rate": 0.015327875046521984,
    "loss": "log_loss",
    "max_depth": 5,
    "max_features": null,
    "max_leaf_nodes": null,
    "min_impurity_decrease": 0.0,
    "min_samples_leaf": 10,
    "min_samples_split": 20,
    "min_weight_fraction_leaf": 0.0,
    "n_estimators": 50,
    "n_iter_no_change": null,
    "random_state": 42,
    "subsample": 1.0,
    "tol": 0.0001,
    "validation_fraction": 0.1,
    "verbose": 0,
    "warm_start": false
  },
  "test_metrics": {
    "accuracy": 0.6512540522095206,
    "precision": 0.6502274795268426,
    "recall": 0.9669824086603518,
    "f1_score": 0.7775843307943416,
    "roc_auc": 0.694657063212122,
    "pr_auc": 0.7943643677015135
  },
  "feature_columns": [
    "Engine_RPM",
    "Lub_Oil_Pressure",
    "Fuel_Pressure",
    "Coolant_Pressure",
    "Lub_Oil_Temperature",
    "Coolant_Temperature"
  ],
  "target_column": "Engine_Condition",
  "class_mapping": {
    "0": "Normal",
    "1": "Faulty"
  }
}