Partial Dependencies

Tools to compute Partial Dependency

orion.analysis.partial_dependency_utils.make_grid(dim, n_points)[source]

Build a grid of n_points for a dim

orion.analysis.partial_dependency_utils.partial_dependency(trials, space, params=None, model='RandomForestRegressor', n_grid_points=10, n_samples=50, **kwargs)[source]

Calculates the partial dependency of parameters in a collection of orion.core.worker.trial.Trial.

Parameters
trials: DataFrame or dict

A dataframe of trials containing, at least, the columns ‘objective’ and ‘id’. Or a dict equivalent.

space: Space object

A space object from an experiment.

params: list of str, optional

The parameters to include in the computation. All parameters are included by default.

model: str

Name of the regression model to use. Can be one of - AdaBoostRegressor - BaggingRegressor - ExtraTreesRegressor - GradientBoostingRegressor - RandomForestRegressor (Default)

n_grid_points: int

Number of points in the grid to compute partial dependency. Default is 10.

n_samples: int

Number of samples to randomly generate the grid used to compute the partial dependency. Default is 50.

**kwargs

Arguments for the regressor model.

Returns
dict

Dictionary of DataFrames. Each combination of parameters as keys (dim1.name, dim2.name) and for each parameters individually (dim1.name). Columns are (dim1.name, dim2.name, objective) or (dim1.name, objective).

orion.analysis.partial_dependency_utils.partial_dependency_grid(space, model, params, samples, n_points=40)[source]

Compute the dependency grid for a given set of params (1 or 2)

orion.analysis.partial_dependency_utils.reverse(transformed_space, grid)[source]

Reverse transformations on the grid to bring back to original space