vastorbit.machine_learning.model_selection.hp_tuning.parameter_grid¶

vastorbit.machine_learning.model_selection.hp_tuning.parameter_grid(param_grid: dict) → list[dict]¶

Generates a list of the different combinations of input parameters.

Parameters:: param_grid (dict) – dictionary of parameters.
Returns:: list of the different combinations.
Return type:: list of dict

Examples

Its easy to generate a list of the different combinations of input parameters.

from vastorbit.machine_learning.model_selection.hp_tuning.param_gen import parameter_grid

parameter_grid(
    {
        "nbins": [10, 100, 360],
        "alpha": [0.1, 0.3, 0.5],
        "solver": ["lbfgs", "newton-cg"],
    },
)

Note

This function is essential for conducting a grid_search_cv().