customer-churn-prediction / training_report.json
itxtx's picture
Upload training_report.json with huggingface_hub
f45cc4c verified
Raw
History Blame Contribute Delete
4.68 kB
{
"generated_at": "2026-06-18T18:07:59",
"best_model": "logistic",
"selection_metric": "f1",
"best_score": 0.632438739789965,
"final_model_path": "models/final_model.joblib",
"models": {
"logistic": {
"model_name": "logistic",
"f1": 0.632438739789965,
"recall": 0.7245989304812834,
"roc_auc": 0.8491487767702601,
"classification_report": {
"No": {
"precision": 0.8887688984881209,
"recall": 0.7951690821256039,
"f1-score": 0.8393676695563488,
"support": 1035.0
},
"Yes": {
"precision": 0.5610766045548654,
"recall": 0.7245989304812834,
"f1-score": 0.632438739789965,
"support": 374.0
},
"accuracy": 0.7764371894960965,
"macro avg": {
"precision": 0.7249227515214931,
"recall": 0.7598840063034437,
"f1-score": 0.7359032046731568,
"support": 1409.0
},
"weighted avg": {
"precision": 0.8017874095377748,
"recall": 0.7764371894960965,
"f1-score": 0.7844411828759885,
"support": 1409.0
}
},
"confusion_matrix": [
[
823,
212
],
[
103,
271
]
]
},
"random_forest": {
"model_name": "random_forest",
"f1": 0.6256860592755215,
"recall": 0.7620320855614974,
"roc_auc": 0.8414567671600919,
"classification_report": {
"No": {
"precision": 0.8979357798165137,
"recall": 0.7565217391304347,
"f1-score": 0.821185107498689,
"support": 1035.0
},
"Yes": {
"precision": 0.5307262569832403,
"recall": 0.7620320855614974,
"f1-score": 0.6256860592755215,
"support": 374.0
},
"accuracy": 0.7579843860894251,
"macro avg": {
"precision": 0.714331018399877,
"recall": 0.7592769123459661,
"f1-score": 0.7234355833871052,
"support": 1409.0
},
"weighted avg": {
"precision": 0.8004649767365675,
"recall": 0.7579843860894251,
"f1-score": 0.7692925283393813,
"support": 1409.0
}
},
"confusion_matrix": [
[
783,
252
],
[
89,
285
]
]
},
"gradient_boosting": {
"model_name": "gradient_boosting",
"f1": 0.6157354618015963,
"recall": 0.7219251336898396,
"roc_auc": 0.8390955591722854,
"classification_report": {
"No": {
"precision": 0.8852097130242825,
"recall": 0.77487922705314,
"f1-score": 0.8263781555899021,
"support": 1035.0
},
"Yes": {
"precision": 0.536779324055666,
"recall": 0.7219251336898396,
"f1-score": 0.6157354618015963,
"support": 374.0
},
"accuracy": 0.7608232789212207,
"macro avg": {
"precision": 0.7109945185399742,
"recall": 0.7484021803714898,
"f1-score": 0.7210568086957492,
"support": 1409.0
},
"weighted avg": {
"precision": 0.7927235771305546,
"recall": 0.7608232789212207,
"f1-score": 0.7704659004608557,
"support": 1409.0
}
},
"confusion_matrix": [
[
802,
233
],
[
104,
270
]
]
},
"xgboost": {
"model_name": "xgboost",
"f1": 0.6223277909738717,
"recall": 0.7005347593582888,
"roc_auc": 0.8408858405022088,
"classification_report": {
"No": {
"precision": 0.8809776833156217,
"recall": 0.8009661835748793,
"f1-score": 0.8390688259109311,
"support": 1035.0
},
"Yes": {
"precision": 0.5598290598290598,
"recall": 0.7005347593582888,
"f1-score": 0.6223277909738717,
"support": 374.0
},
"accuracy": 0.7743080198722498,
"macro avg": {
"precision": 0.7204033715723408,
"recall": 0.750750471466584,
"f1-score": 0.7306983084424015,
"support": 1409.0
},
"weighted avg": {
"precision": 0.7957331232134399,
"recall": 0.7743080198722498,
"f1-score": 0.7815378485749054,
"support": 1409.0
}
},
"confusion_matrix": [
[
829,
206
],
[
112,
262
]
]
}
}
}