| { | |
| "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" | |
| ] | |
| } |