model_comparison.json

{
    "random_forest": {
        "model": "random_forest",
        "best_params": {
            "estimator__max_depth": 12,
            "estimator__min_samples_split": 9,
            "estimator__n_estimators": 166
        },
        "cv_best_score": 0.7668745851009632,
        "test_r2": 0.8752436796179663,
        "test_rmse": 1830.462659704419
    },
    "xgboost": {
        "model": "xgboost",
        "params": {
            "n_estimators": 100,
            "max_depth": 6,
            "learning_rate": 0.1,
            "subsample": 0.8
        },
        "test_r2": 0.8900634418034745,
        "test_rmse": 1718.3066213876568
    }
}