add a few models

closes #28

Author: penelopeysm

main.jl

@@ -44,6 +44,7 @@ include("models/assume_mvnormal.jl")
 include("models/assume_normal.jl")
 include("models/assume_submodel.jl")
 include("models/assume_wishart.jl")
+include("models/broadcast_macro.jl")
 include("models/control_flow.jl")
 include("models/demo_assume_dot_observe.jl")
 include("models/demo_assume_dot_observe_literal.jl")
@@ -67,6 +68,8 @@ include("models/n010.jl")
 include("models/n050.jl")
 include("models/n100.jl")
 include("models/n500.jl")
+include("models/observe_bernoulli.jl")
+include("models/observe_categorical.jl")
 include("models/observe_index.jl")
 include("models/observe_literal.jl")
 include("models/observe_multivariate.jl")

models/broadcast_macro.jl

@@ -0,0 +1,10 @@
+@model function broadcast_macro(
+    x = [1.5, 2.0],
+    ::Type{TV} = Vector{Float64},
+) where {TV}
+    a ~ Normal(0, 1)
+    b ~ InverseGamma(2, 3)
+    @. x ~ Normal(a, $(sqrt(b)))
+end
+
+@register broadcast_macro()

models/observe_bernoulli.jl

@@ -0,0 +1,8 @@
+@model function observe_bernoulli(x = [true, false, true])
+    a ~ Beta(2, 2)
+    for i in eachindex(x)
+        x[i] ~ Bernoulli(a)
+    end
+end
+
+@register observe_bernoulli()

models/observe_categorical.jl

@@ -0,0 +1,8 @@
+@model function observe_categorical(x = [1, 2, 1, 2, 2])
+    a ~ Dirichlet(2, 3)
+    for i in eachindex(x)
+        x[i] ~ Categorical(a)
+    end
+end
+
+@register observe_categorical()