File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -116,12 +116,13 @@ def dic(trace, model=None):
116
116
`float` representing the deviance information criterion of the model and trace
117
117
"""
118
118
model = modelcontext (model )
119
+ logp = model .logp
119
120
120
- mean_deviance = - 2 * np .mean ([model . logp (pt ) for pt in trace ])
121
+ mean_deviance = - 2 * np .mean ([logp (pt ) for pt in trace ])
121
122
122
123
free_rv_means = {rv .name : trace [rv .name ].mean (
123
124
axis = 0 ) for rv in model .free_RVs }
124
- deviance_at_mean = - 2 * model . logp (free_rv_means )
125
+ deviance_at_mean = - 2 * logp (free_rv_means )
125
126
126
127
return 2 * mean_deviance - deviance_at_mean
127
128
@@ -328,12 +329,13 @@ def bpic(trace, model=None):
328
329
Optional model. Default None, taken from context.
329
330
"""
330
331
model = modelcontext (model )
332
+ logp = model .logp
331
333
332
- mean_deviance = - 2 * np .mean ([model . logp (pt ) for pt in trace ])
334
+ mean_deviance = - 2 * np .mean ([logp (pt ) for pt in trace ])
333
335
334
336
free_rv_means = {rv .name : trace [rv .name ].mean (
335
337
axis = 0 ) for rv in model .free_RVs }
336
- deviance_at_mean = - 2 * model . logp (free_rv_means )
338
+ deviance_at_mean = - 2 * logp (free_rv_means )
337
339
338
340
return 3 * mean_deviance - 2 * deviance_at_mean
339
341
You can’t perform that action at this time.
0 commit comments