@@ -141,17 +141,16 @@ end
141
141
return y ~ MvNormal (mu, error^ 2 * I)
142
142
end
143
143
144
- # Some data
145
144
x = randn (2 , 100 )
146
145
y = [1 + 2 * a + 3 * b for (a, b) in eachcol (x)]
147
146
148
147
param_names = Dict (
149
- simple_linear1 => [:intercept , : coef ],
150
- simple_linear2 => [:intercept , : coef ],
151
- simple_linear3 => [:intercept , Symbol .([ " coef[$i ] " for i in 1 : 2 ]) ... ],
152
- simple_linear4 => [:intercept , :coef1 , :coef2 ],
148
+ simple_linear1 => [:intercept , Symbol ( " coef[1] " ), Symbol ( " coef[2] " ), :error ],
149
+ simple_linear2 => [:intercept , Symbol ( " coef[1] " ), Symbol ( " coef[2] " ), :error ],
150
+ simple_linear3 => [:intercept , Symbol ( " coef[1] " ), Symbol ( " coef[2] " ), :error ],
151
+ simple_linear4 => [:intercept , :coef1 , :coef2 , :error ],
153
152
)
154
- for model in [simple_linear1, simple_linear2, simple_linear3, simple_linear4]
153
+ @testset " $model " for model in [simple_linear1, simple_linear2, simple_linear3, simple_linear4]
155
154
m = model (x, y)
156
155
chain = sample (
157
156
DynamicPPL. LogDensityFunction (m, DynamicPPL. VarInfo (m)),
0 commit comments