| { | |
| "RandomForest": { | |
| "cv_roc_auc_mean": 0.6570736663913218, | |
| "cv_roc_auc_std": 0.23626271641720448, | |
| "best_params": { | |
| "model__n_estimators": 1000, | |
| "model__min_samples_split": 5, | |
| "model__min_samples_leaf": 1, | |
| "model__max_features": null, | |
| "model__max_depth": null | |
| } | |
| }, | |
| "XGBoost": { | |
| "cv_roc_auc_mean": 0.6802915502368345, | |
| "cv_roc_auc_std": 0.25513056403399653, | |
| "best_params": { | |
| "model__subsample": 1.0, | |
| "model__scale_pos_weight": 50, | |
| "model__n_estimators": 500, | |
| "model__max_depth": 5, | |
| "model__learning_rate": 0.2, | |
| "model__colsample_bytree": 1.0 | |
| } | |
| }, | |
| "LightGBM": { | |
| "cv_roc_auc_mean": 0.7117485460646773, | |
| "cv_roc_auc_std": 0.23671957245320416, | |
| "best_params": { | |
| "model__subsample": 0.6, | |
| "model__num_leaves": 100, | |
| "model__n_estimators": 1000, | |
| "model__max_depth": 3, | |
| "model__learning_rate": 0.01, | |
| "model__colsample_bytree": 0.6 | |
| } | |
| } | |
| } |