Skip to content

glassbox.orchestrator.grid_search

GridSearchCV for exhaustive search.


GridSearchCV

GridSearchCV(
    estimator, param_space, cv_engine, scoring_func
)

Bases: BaseSearch

Exhaustive grid search over a parameter space.

Parameters:

Name Type Description Default
estimator BaseModel

The model to optimize.

required
param_space Dict

Parameter grid for exhaustive search.

required
cv_engine BaseSplitter

Cross-validation splitter.

required
scoring_func Callable

Scoring function used to evaluate candidates.

required
Source code in glassbox/orchestrator/base_search.py
def __init__(
    self,
    estimator: BaseModel,
    param_space: Dict,
    cv_engine: BaseSplitter,
    scoring_func: Callable,
) -> None:
    self.estimator: BaseModel = estimator
    self.param_space: Dict = param_space
    self.cv_engine: BaseSplitter = cv_engine
    self.scoring_func: Callable = scoring_func
    self.best_params_: Dict = {}
    self.best_score_: float = 0.0
    self.best_estimator_: BaseModel = estimator