aplicar_transformador_sklearn_a_lo_largo_de_dim

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

Aplique un transformador de scikit-learn a través de una dimensión de un xarray DataArray.

Función auxiliar para utilizar las funciones de scikit-learn con el objetivo de xarray.

Parámetros:

datos : xr.DataArrayxr.DataArray

Los datos de entrada a transformar.

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

método de scikit-learn para aplicar a los datos

dim_namestr

Nombre de la dimensión a la que se aplicará la función

Devoluciones:

xr.DataArray