Skip to content

Commit 56689eb

Browse files
author
colin
committed
Update notebooks with current API, and add kwarg to tracetab
1 parent 6f0013f commit 56689eb

10 files changed

+945
-792
lines changed

docs/source/notebooks/BEST.ipynb

+49-37
Large diffs are not rendered by default.

docs/source/notebooks/GLM-model-selection.ipynb

+311-83
Large diffs are not rendered by default.

docs/source/notebooks/GLM.ipynb

+208-119
Large diffs are not rendered by default.

docs/source/notebooks/advi.ipynb

+61-38
Large diffs are not rendered by default.

docs/source/notebooks/disaster_model.ipynb

+36-114
Large diffs are not rendered by default.

docs/source/notebooks/gaussian_mixture_model.ipynb

+44-28
Large diffs are not rendered by default.

docs/source/notebooks/profiling.ipynb

+160-134
Large diffs are not rendered by default.

docs/source/notebooks/rugby_analytics.ipynb

+35-28
Large diffs are not rendered by default.

docs/source/notebooks/tutorial.ipynb

+38-208
Large diffs are not rendered by default.

pymc3/backends/tracetab.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
__all__ = ['trace_to_dataframe']
88

99

10-
def trace_to_dataframe(trace, chains=None, flat_names=None):
10+
def trace_to_dataframe(trace, chains=None, flat_names=None, hide_transformed_vars=True):
1111
"""Convert trace to Pandas DataFrame.
1212
1313
Parameters
@@ -23,11 +23,11 @@ def trace_to_dataframe(trace, chains=None, flat_names=None):
2323
if flat_names is None:
2424
flat_names = {v: create_flat_names(v, shape)
2525
for v, shape in var_shapes.items()
26-
if not v.endswith('_')}
26+
if not hide_transformed_vars or not v.endswith('_')}
2727

2828
var_dfs = []
2929
for varname, shape in var_shapes.items():
30-
if not varname.endswith('_'):
30+
if not hide_transformed_vars or not varname.endswith('_'):
3131
vals = trace.get_values(varname, combine=True, chains=chains)
3232
flat_vals = vals.reshape(vals.shape[0], -1)
3333
var_dfs.append(pd.DataFrame(flat_vals, columns=flat_names[varname]))

0 commit comments

Comments
 (0)