BuildModelFromDAG.build

BuildModelFromDAG.build()

Construya y devuelva el modelo PyMC implícito en el DAG y los datos.

El método crea un contenedor pm.Data para cada nodo con el fin de alinear los datos observados con los dims declarados. Para cada arista A -> B, se instancia un prior de pendiente a partir de model_config['slope'] y se utiliza en la media de la verosimilitud del nodo B, que se instancia a partir de model_config['likelihood'].

Devoluciones:

pymc.Model

Un modelo completamente especificado con pendientes y probabilidades para todos los nodos.

Ejemplos

Construya un modelo y muestree de él:

builder = BuildModelFromDAG(
    dag="A->B", df=df, target="B", dims=("date",), coords={"date": dates}
)
model = builder.build()
with model:
    idata = pm.sample(100, tune=100, chains=2, cores=2)

Dimensiones multidimensionales (por ejemplo, fecha y país):

dims = ("date", "country")
coords = {"date": dates, "country": ["Venezuela", "Colombia"]}
builder = BuildModelFromDAG(
    dag="A->B, B->Y", df=df, target="Y", dims=dims, coords=coords
)
model = builder.build()