metadata.json

{
  "model_name": "LightGBM",
  "dataset": "mascalmeida/industrial_machine_predictive_maintenance_classification",
  "task": "tabular-classification",
  "features": [
    "Air temperature [K]",
    "Process temperature [K]",
    "Rotational speed [rpm]",
    "Torque [Nm]",
    "Tool wear [min]",
    "Type_encoded",
    "temp_diff",
    "power_proxy",
    "torque_wear",
    "speed_wear",
    "temp_torque"
  ],
  "target": "Machine failure",
  "metrics": {
    "macro_f1": 0.8919279494819063,
    "weighted_f1": 0.9856508415645107,
    "auc_roc": 0.9601677627572769,
    "accuracy": 0.9855,
    "precision": 0.7746478873239436,
    "recall": 0.8088235294117647
  },
  "cv_results": {
    "LightGBM": {
      "f1_macro": [
        0.8863075178460091,
        0.006877717214825891
      ],
      "f1_weighted": [
        0.9850516108630073,
        0.0008163649368650066
      ],
      "auc_roc": [
        0.9680226062741701,
        0.00554719735462247
      ],
      "accuracy": [
        0.9850000000000001,
        0.0008366600265340631
      ],
      "precision": [
        0.776502647127647,
        0.03005446722702152
      ],
      "recall": [
        0.7876646180860404,
        0.04493691968805627
      ]
    },
    "XGBoost": {
      "f1_macro": [
        0.7319899731098649,
        0.01220999584077537
      ],
      "f1_weighted": [
        0.952195887243394,
        0.0033972044669525196
      ],
      "auc_roc": [
        0.9555805933233275,
        0.009845449038815925
      ],
      "accuracy": [
        0.9401999999999999,
        0.005055689863905814
      ],
      "precision": [
        0.34795307204793224,
        0.020997250668425647
      ],
      "recall": [
        0.8642669007901669,
        0.01747047349834227
      ]
    },
    "RandomForest": {
      "f1_macro": [
        0.7804095750299703,
        0.023757470025984433
      ],
      "f1_weighted": [
        0.9645414912596542,
        0.00531703853026116
      ],
      "auc_roc": [
        0.971268927553813,
        0.00585556061341795
      ],
      "accuracy": [
        0.9581,
        0.007398648525237564
      ],
      "precision": [
        0.44499278840132106,
        0.05102892352219316
      ],
      "recall": [
        0.852502194907814,
        0.024623305046840353
      ]
    }
  },
  "preprocessing": "StandardScaler + SMOTE (in-fold)",
  "feature_engineering": [
    "temp_diff = Air temp - Process temp",
    "power_proxy = Torque / (Rotational speed + 1)",
    "torque_wear = Torque \u00d7 Tool wear",
    "speed_wear = Rotational speed \u00d7 Tool wear",
    "temp_torque = Process temp \u00d7 Torque"
  ]
}