tourism-package-model / model_metadata.json
DD009's picture
Upload model_metadata.json with huggingface_hub
13191d4 verified
{
"model_type": "XGBoost",
"best_params": {
"colsample_bytree": 0.5,
"learning_rate": 0.2,
"max_depth": 7,
"n_estimators": 200,
"reg_lambda": 1.5
},
"train_metrics": {
"accuracy": 1.0,
"recall": 1.0,
"f1_score": 1.0,
"roc_auc": 1.0
},
"test_metrics": {
"accuracy": 0.9382566585956417,
"recall": 0.7924528301886793,
"f1_score": 0.8316831683168316,
"roc_auc": 0.9682045769568046
},
"class_weight": 4.175548589341693,
"features": [
"Unnamed: 0",
"Age",
"TypeofContact",
"CityTier",
"DurationOfPitch",
"Occupation",
"Gender",
"NumberOfPersonVisiting",
"NumberOfFollowups",
"ProductPitched",
"PreferredPropertyStar",
"MaritalStatus",
"NumberOfTrips",
"Passport",
"PitchSatisfactionScore",
"OwnCar",
"NumberOfChildrenVisiting",
"Designation",
"MonthlyIncome"
]
}