Skip to content

Commit 810cc27

Browse files
committed
hide transformed variable from user
1 parent 5317dd4 commit 810cc27

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pymc3/backends/tracetab.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,13 +22,15 @@ def trace_to_dataframe(trace, chains=None, flat_names=None):
2222
var_shapes = trace._straces[0].var_shapes
2323
if flat_names is None:
2424
flat_names = {v: create_flat_names(v, shape)
25-
for v, shape in var_shapes.items()}
25+
for v, shape in var_shapes.items()
26+
if not v.endswith('_')}
2627

2728
var_dfs = []
2829
for varname, shape in var_shapes.items():
29-
vals = trace.get_values(varname, combine=True, chains=chains)
30-
flat_vals = vals.reshape(vals.shape[0], -1)
31-
var_dfs.append(pd.DataFrame(flat_vals, columns=flat_names[varname]))
30+
if not varname.endswith('_'):
31+
vals = trace.get_values(varname, combine=True, chains=chains)
32+
flat_vals = vals.reshape(vals.shape[0], -1)
33+
var_dfs.append(pd.DataFrame(flat_vals, columns=flat_names[varname]))
3234
return pd.concat(var_dfs, axis=1)
3335

3436

0 commit comments

Comments
 (0)