Add full DynamicPPL demo models

Author: penelopeysm

main.jl

@@ -44,8 +44,9 @@ include("models/assume_normal.jl")
 include("models/assume_submodel.jl")
 include("models/assume_wishart.jl")
 include("models/control_flow.jl")
-include("models/dot_assume_observe_index.jl")
+include("models/demo_dot_assume_observe_index.jl")
 include("models/dot_assume.jl")
+include("models/demo_dot_assume_observe.jl")
 include("models/dot_observe.jl")
 include("models/dynamic_constraint.jl")
 include("models/multiple_constraints_same_var.jl")

models/demo_assume_dot_observe.jl

@@ -0,0 +1,10 @@
+@model function demo_assume_dot_observe(x=[1.5, 2.0])
+    # `assume` and `dot_observe`
+    s ~ InverseGamma(2, 3)
+    m ~ Normal(0, sqrt(s))
+    x .~ Normal(m, sqrt(s))
+
+    return (; s=s, m=m, x=x, logp=getlogp(__varinfo__))
+end
+
+@register demo_assume_dot_observe()

models/demo_assume_dot_observe_literal.jl

@@ -0,0 +1,10 @@
+@model function demo_assume_dot_observe_literal()
+    # `assume` and literal `dot_observe`
+    s ~ InverseGamma(2, 3)
+    m ~ Normal(0, sqrt(s))
+    [1.5, 2.0] .~ Normal(m, sqrt(s))
+
+    return (; s=s, m=m, x=[1.5, 2.0], logp=getlogp(__varinfo__))
+end
+
+@model demo_assume_dot_observe_literal()

models/demo_assume_index_observe.jl

@@ -0,0 +1,19 @@
+@model function demo_assume_index_observe(
+    x = [1.5, 2.0],
+    ::Type{TV} = Vector{Float64},
+) where {TV}
+    # `assume` with indexing and `observe`
+    s = TV(undef, length(x))
+    for i in eachindex(s)
+        s[i] ~ InverseGamma(2, 3)
+    end
+    m = TV(undef, length(x))
+    for i in eachindex(m)
+        m[i] ~ Normal(0, sqrt(s[i]))
+    end
+    x ~ MvNormal(m, Diagonal(s))
+
+    return (; s = s, m = m, x = x, logp = getlogp(__varinfo__))
+end
+
+@register demo_assume_index_observe()

models/demo_assume_matrix_observe_matrix_index.jl

@@ -0,0 +1,15 @@
+@model function demo_assume_matrix_observe_matrix_index(
+    x=transpose([1.5 2.0;]), ::Type{TV}=Array{Float64}
+) where {TV}
+    n = length(x)
+    d = n ÷ 2
+    s ~ reshape(product_distribution(fill(InverseGamma(2, 3), n)), d, 2)
+    s_vec = vec(s)
+    m ~ MvNormal(zeros(n), Diagonal(s_vec))
+
+    x[:, 1] ~ MvNormal(m, Diagonal(s_vec))
+
+    return (; s=s, m=m, x=x, logp=getlogp(__varinfo__))
+end
+
+@register demo_assume_matrix_observe_matrix_index()

models/demo_assume_multivariate_observe.jl

@@ -0,0 +1,10 @@
+@model function demo_assume_multivariate_observe(x=[1.5, 2.0])
+    # Multivariate `assume` and `observe`
+    s ~ product_distribution([InverseGamma(2, 3), InverseGamma(2, 3)])
+    m ~ MvNormal(zero(x), Diagonal(s))
+    x ~ MvNormal(m, Diagonal(s))
+
+    return (; s=s, m=m, x=x, logp=getlogp(__varinfo__))
+end
+
+@register demo_assume_multivariate_observe()

models/demo_assume_multivariate_observe_literal.jl

@@ -0,0 +1,10 @@
+@model function demo_assume_multivariate_observe_literal()
+    # multivariate `assume` and literal `observe`
+    s ~ product_distribution([InverseGamma(2, 3), InverseGamma(2, 3)])
+    m ~ MvNormal(zeros(2), Diagonal(s))
+    [1.5, 2.0] ~ MvNormal(m, Diagonal(s))
+
+    return (; s=s, m=m, x=[1.5, 2.0], logp=getlogp(__varinfo__))
+end
+
+@register demo_assume_multivariate_observe_literal()

models/demo_assume_observe_literal.jl

@@ -0,0 +1,11 @@
+@model function demo_assume_observe_literal()
+    # univariate `assume` and literal `observe`
+    s ~ InverseGamma(2, 3)
+    m ~ Normal(0, sqrt(s))
+    1.5 ~ Normal(m, sqrt(s))
+    2.0 ~ Normal(m, sqrt(s))
+
+    return (; s=s, m=m, x=[1.5, 2.0], logp=getlogp(__varinfo__))
+end
+
+@register demo_assume_observe_literal()

models/demo_assume_submodel_observe_index_literal.jl

@@ -0,0 +1,19 @@
+@model function _prior_dot_assume(::Type{TV}=Vector{Float64}) where {TV}
+    s = TV(undef, 2)
+    s .~ InverseGamma(2, 3)
+    m = TV(undef, 2)
+    m ~ product_distribution(Normal.(0, sqrt.(s)))
+    return s, m
+end
+
+@model function demo_assume_submodel_observe_index_literal()
+    # Submodel prior
+    priors ~ to_submodel(_prior_dot_assume(), false)
+    s, m = priors
+    1.5 ~ Normal(m[1], sqrt(s[1]))
+    2.0 ~ Normal(m[2], sqrt(s[2]))
+
+    return (; s=s, m=m, x=[1.5, 2.0], logp=getlogp(__varinfo__))
+end
+
+@register demo_assume_submodel_observe_index_literal()

models/demo_dot_assume_observe.jl

@@ -0,0 +1,15 @@
+@model function demo_dot_assume_observe(
+    x = [1.5, 2.0],
+    ::Type{TV} = Vector{Float64},
+) where {TV}
+    # `dot_assume` and `observe`
+    s = TV(undef, length(x))
+    m = TV(undef, length(x))
+    s .~ InverseGamma(2, 3)
+    m ~ product_distribution(Normal.(0, sqrt.(s)))
+
+    x ~ MvNormal(m, Diagonal(s))
+    return (; s = s, m = m, x = x, logp = getlogp(__varinfo__))
+end
+
+@register demo_dot_assume_observe()