Skip to content

Commit 5e92892

Browse files
authored
add mlflow error docs (#1616)
1 parent 06439db commit 5e92892

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

pymc_marketing/mlflow.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
- :func:`log_sample_diagnostics`: Log information derived from the InferenceData object.
3030
- :func:`log_arviz_summary`: Log table of summary statistics about estimated parameters
3131
- :func:`log_metadata`: Log the metadata of the data used in the model.
32+
- :func:`log_error`: Log the exception if an error occurs during sampling.
3233
- `pymc.find_MAP`:
3334
- :func:`log_model_derived_info`: Log types of parameters, coords, model graph, etc.
3435
- `MMM.fit`:
@@ -1057,6 +1058,9 @@ def autolog(
10571058
Whether to log the metadata of inputs used in the model. Default is True.
10581059
log_model_info : bool, optional
10591060
Whether to log model information. Default is True.
1061+
sample_error_file : str, optional
1062+
The name of the file to log the error if an error occurs during sampling. If
1063+
None, the error will not be logged. Default is "sample-error.txt".
10601064
summary_var_names : list[str], optional
10611065
The names of the variables to include in the ArviZ summary. Default is
10621066
all the variables in the InferenceData object.

0 commit comments

Comments
 (0)