edaflow.ml.grid_search_models
- edaflow.ml.grid_search_models(models: Dict[str, BaseEstimator], param_grids: Dict[str, Dict[str, Any]], X_train: DataFrame, y_train: Series, cv: int = 5, scoring: str = 'auto', verbose: bool = True) Dict[str, Dict[str, Any]][source]
Perform grid search optimization for multiple models.
Parameters:
- modelsDict[str, BaseEstimator]
Dictionary of model name -> model pairs
- param_gridsDict[str, Dict[str, Any]]
Dictionary of model name -> parameter grid pairs
- X_trainpd.DataFrame
Training features
- y_trainpd.Series
Training target
- cvint, default=5
Number of cross-validation folds
- scoringstr, default=’auto’
Scoring metric
- verbosebool, default=True
Whether to print progress
Returns:
- Dict[str, Dict[str, Any]]
Dictionary of model name -> optimization results pairs