crear_indexador_de_variables#

pymc_marketing.mmm.lift_test.create_variable_indexer(model, indices)[fuente]#

Cree una función para indexar variables en el modelo.

Parámetros:

modelo : pm.Modelpm.Model: Modelo PyMC
índices : dict[str, np.ndarray]dict[str, np.ndarray]: Diccionario de índices para los índices en el modelo.

Devoluciones:

Callable[[str], TensorVariable]: Función para indexar variables en el modelo.

Aumentos:

KeyError: Si la variable no está en el modelo.

Ejemplos

Crear un indexador de variables:

import numpy as np
import pymc as pm

from pymc_marketing.mmm.lift_test import create_variable_indexer

coords = {"channel": [0, 1, 2], "geo": ["A", "B", "C"]}
with pm.Model(coords=coords) as model:
    pm.Normal("alpha", dims=("channel", "geo"))
    pm.Normal("beta", dims="channel")

# Usually from exact_row_indices
indices = {"channel": [0, 1], "geo": [1, 0]}

variable_indexer = create_variable_indexer(model, indices)

Obtener alfa en los índices:

alpha_at_indices = variable_indexer("alpha")

Obtener beta en índices:

beta_at_indices = variable_indexer("beta")