wellness-classifier-model / tuning_results_summary.json
sathishaiuse's picture
Upload tuning_results_summary.json with huggingface_hub
9730d3f verified
{
"DecisionTree": {
"best_params": {
"clf__max_depth": null,
"clf__min_samples_split": 2
},
"cv_best_score": 0.6300000794885704,
"test_metrics": {
"accuracy": 0.8966376089663761,
"precision": 0.7608695652173914,
"recall": 0.6774193548387096,
"f1": 0.7167235494880545
}
},
"Bagging": {
"best_params": {
"clf__max_samples": 1.0,
"clf__n_estimators": 50
},
"cv_best_score": 0.7074632721981063,
"test_metrics": {
"accuracy": 0.9227895392278954,
"precision": 0.8907563025210085,
"recall": 0.6838709677419355,
"f1": 0.7737226277372262
}
},
"RandomForest": {
"best_params": {
"clf__max_depth": null,
"clf__n_estimators": 100
},
"cv_best_score": 0.6475210405162785,
"test_metrics": {
"accuracy": 0.9041095890410958,
"precision": 0.90625,
"recall": 0.5612903225806452,
"f1": 0.6932270916334662
}
},
"AdaBoost": {
"best_params": {
"clf__learning_rate": 1.0,
"clf__n_estimators": 50
},
"cv_best_score": 0.4779972612213404,
"test_metrics": {
"accuracy": 0.825653798256538,
"precision": 0.6027397260273972,
"recall": 0.2838709677419355,
"f1": 0.38596491228070173
}
},
"GradientBoosting": {
"best_params": {
"clf__learning_rate": 0.1,
"clf__n_estimators": 100
},
"cv_best_score": 0.5438301270764477,
"test_metrics": {
"accuracy": 0.8617683686176837,
"precision": 0.782051282051282,
"recall": 0.3935483870967742,
"f1": 0.5236051502145923
}
},
"XGBoost": {
"best_params": {
"clf__learning_rate": 0.1,
"clf__max_depth": 6,
"clf__n_estimators": 100
},
"cv_best_score": 0.7289349713719079,
"test_metrics": {
"accuracy": 0.9041095890410958,
"precision": 0.74375,
"recall": 0.7677419354838709,
"f1": 0.7555555555555556
}
}
}