log_arviz_summary#

pymc_marketing.mlflow.log_arviz_summary(idata, path, var_names=None, **summary_kwargs)[source]#

Log the ArviZ summary as an artifact on MLflow.

Automatically removes the file after logging.

Parameters:
idataaz.InferenceData

The InferenceData object returned by the sampling method.

pathstr | Path

The path to save the summary as HTML.

var_nameslist[str], optional

The names of the variables to include in the summary. Default is all the variables in the InferenceData object.

summary_kwargsdict

Additional keyword arguments to pass to az.summary.