Prior.to_graph#

Prior.to_graph()[source]#

Generate a graph of the variables.

Examples

Create the graph for a 2D transformed hierarchical distribution.

from pymc_marketing.prior import Prior

mu = Prior(
    "Normal",
    mu=Prior("Normal"),
    sigma=Prior("HalfNormal"),
    dims="channel",
)
sigma = Prior("HalfNormal", dims="channel")
dist = Prior(
    "Normal",
    mu=mu,
    sigma=sigma,
    dims=("channel", "geo"),
    centered=False,
    transform="sigmoid",
)

dist.to_graph()