predictive-maintenance / model_meta.json
subratm62's picture
Upload model_meta.json with huggingface_hub
1f7de89 verified
{
"threshold": 0.5,
"best_params": {
"gbm__subsample": 0.7,
"gbm__n_estimators": 300,
"gbm__min_samples_split": 20,
"gbm__min_samples_leaf": 8,
"gbm__max_features": null,
"gbm__max_depth": 3,
"gbm__learning_rate": 0.05
},
"cv_f1_macro": 0.613859570195129,
"train_metrics": {
"0": {
"precision": 0.6557169930511687,
"recall": 0.44944793245291187,
"f1-score": 0.5333333333333333,
"support": 4619.0
},
"1": {
"precision": 0.7276135389888603,
"recall": 0.8617277686160091,
"f1-score": 0.7890121377548057,
"support": 7883.0
},
"accuracy": 0.7094064949608063,
"macro avg": {
"precision": 0.6916652660200144,
"recall": 0.6555878505344604,
"f1-score": 0.6611727355440695,
"support": 12502.0
},
"weighted avg": {
"precision": 0.7010505774078175,
"recall": 0.7094064949608063,
"f1-score": 0.6945488200758119,
"support": 12502.0
}
},
"test_metrics": {
"0": {
"precision": 0.5816115702479339,
"recall": 0.3898891966759003,
"f1-score": 0.46683250414593697,
"support": 1444.0
},
"1": {
"precision": 0.7002381762504253,
"recall": 0.8355663824604141,
"f1-score": 0.7619400222139948,
"support": 2463.0
},
"accuracy": 0.670847197338111,
"macro avg": {
"precision": 0.6409248732491796,
"recall": 0.6127277895681572,
"f1-score": 0.6143862631799659,
"support": 3907.0
},
"weighted avg": {
"precision": 0.6563946085341218,
"recall": 0.670847197338111,
"f1-score": 0.6528703380342468,
"support": 3907.0
}
},
"features": [
"Engine rpm",
"Lub oil pressure",
"Fuel pressure",
"Coolant pressure",
"lub oil temp",
"Coolant temp"
]
}