log_sample_diagnostics

pymc_marketing.mlflow.log_sample_diagnostics(idata, tune=None)

Log sample diagnostics to MLflow.

Includes:

• The total number of divergences

• The total sampling time in seconds (if available)

• The time per draw in seconds (if available)

• The number of tuning steps (if available)

• The number of draws

• The number of chains

• The inference library used

• The version of the inference library

• The version of ArviZ

Parameters:

idataaz.InferenceData

The InferenceData object returned by the sampling method.

tuneint, optional

The number of tuning steps used in sampling. Derived from the inference data if not provided.