cs-pred-maintain-model / model_metadata.json
spac1ngcat's picture
Upload folder using huggingface_hub
a558c12 verified
{
"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"
}
}