tabensemb.trainer.Trainer._generate_grid#

method

Trainer._generate_grid(feature: str, grid_size: int, percentile: int | float = 100, x_min: int | float | None = None, x_max: int | float | None = None, df: DataFrame | None = None) ndarray[source]#

Generate a sequential (linspace) grid for a feature in the tabular dataset.

Parameters:
feature

The focused feature.

grid_size

The number of sequential values.

percentile

The percentile of the feature used to generate sequential values.

x_min

The lower limit of the generated sequential values. It will override the left percentile.

x_max

The upper limit of the generated sequential values. It will override the right percentile.

df

The tabular dataset.

Returns:
np.ndarray