damuta.callbacks module

class Callback

Bases: object

class LogELBO(every=100)

Bases: Callback

Log ELBO using wandb.log(). wandb.init() must be run first.

Parameters:

every (int) – Frequency at which wandb.log() is called

Examples

>>> with model:
...     approx = pm.fit(n=1000, callbacks=[LogELBO(every=50)])