apply_sklearn_transformer_across_dim

pymc_marketing.mmm.utils.apply_sklearn_transformer_across_dim(data, func, dim_name)

Apply a scikit-learn transformer across a dimension of an xarray DataArray.

Helper function in order to use scikit-learn functions with the xarray target.

Parameters:

dataxr.DataArray

The input data to transform.

funcCallable[[np.ndarray], np.ndarray]

scikit-learn method to apply to the data

dim_namestr

Name of the dimension to apply the function to

Returns:

xr.DataArray