tourism-project-model / model_summary.json
abhisekbasu's picture
Upload model_summary.json with huggingface_hub
b6601fe verified
{
"best_params": {
"xgbclassifier__colsample_bytree": 1.0,
"xgbclassifier__learning_rate": 0.1,
"xgbclassifier__max_depth": 4,
"xgbclassifier__n_estimators": 120,
"xgbclassifier__reg_lambda": 0.5,
"xgbclassifier__subsample": 0.8
},
"threshold": 0.45,
"best_cv_f1": 0.6247849348397656,
"metrics": {
"train": {
"accuracy": 0.9400363416111448,
"precision_1": 0.921455938697318,
"recall_1": 0.7539184952978056,
"f1_1": 0.8293103448275863,
"roc_auc": 0.9734160100539411,
"confusion_matrix": [
[
2623,
41
],
[
157,
481
]
]
},
"test": {
"accuracy": 0.8946731234866828,
"precision_1": 0.8157894736842105,
"recall_1": 0.5849056603773585,
"f1_1": 0.6813186813186813,
"roc_auc": 0.9075085098959954,
"confusion_matrix": [
[
646,
21
],
[
66,
93
]
]
}
}
}