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