apply_sklearn_transformer_across_dim

pymc_marketing.mmm.utils.apply_sklearn_transformer_across_dim(data, func, dim_name, combined=False)

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

Parameters:

• data (DataArray)

• func (scikit-learn method to apply to the data)

• dim_name (Name of the dimension to apply the function to)

• combined (Flag to indicate if the data coords have been combined or not)

Return type:

xr.DataArray