@@ -203,53 +203,6 @@ def test_run(self):
203
203
pm .glm .plot_posterior_predictive (trace )
204
204
205
205
206
- class TestHierarchical (SeededTest ):
207
- @classmethod
208
- def setUpClass (cls ):
209
- n_groups = 10
210
- no_pergroup = 30
211
- n_group_predictors = 1
212
- n_predictors = 3
213
- n_observed = no_pergroup * n_groups
214
-
215
- group = np .concatenate ([[i ] * no_pergroup for i in range (n_groups )])
216
- group_predictors = np .random .normal (size = (n_groups , n_group_predictors ))
217
- predictors = np .random .normal (size = (n_observed , n_predictors ))
218
-
219
- group_effects_a = np .random .normal (size = (n_group_predictors , n_predictors ))
220
- effects_a = (np .random .normal (size = (n_groups , n_predictors )) +
221
- np .dot (group_predictors , group_effects_a ))
222
-
223
- y = np .sum (effects_a [group , :] * predictors , 1 ) + np .random .normal (size = (n_observed ))
224
- with pm .Model () as cls .model :
225
- # m_g ~ N(0, .1)
226
- group_effects = pm .Normal ("group_effects" , 0 , .1 ,
227
- shape = (1 , n_group_predictors , n_predictors ))
228
- # sg ~ Uniform(.05, 10)
229
- sg = pm .Uniform ("sg" , .05 , 10 , testval = 2. )
230
- # m ~ N(mg * pg, sg)
231
- effects = pm .Normal ("effects" ,
232
- (group_predictors [:, :, np .newaxis ] * group_effects ).sum (),
233
- sg ** - 2 ,
234
- shape = (n_groups , n_predictors ))
235
- s = pm .Uniform ("s" , .01 , 10 , shape = n_groups )
236
- g = tt .constant (group )
237
- # y ~ Normal(m[g] * p, s)
238
- pm .Normal ('y' , (effects [g ] * predictors ).sum (), s [g ] ** - 2 , observed = y )
239
-
240
- def test_normal (self ):
241
- with self .model :
242
- start = pm .find_MAP ()
243
- step = pm .NUTS (self .model .vars , scaling = start )
244
- pm .sample (50 , step , start )
245
-
246
- def test_sqlite (self ):
247
- with self .model :
248
- start = pm .find_MAP ()
249
- step = pm .NUTS (self .model .vars , scaling = start )
250
- pm .sample (50 , step , start , trace = 'sqlite' )
251
-
252
-
253
206
class TestLatentOccupancy (SeededTest ):
254
207
"""
255
208
From the PyMC example list
0 commit comments