File size: 904 Bytes
13191d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | {
"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"
]
} |