Implement StatsBase.predict

Author: sethaxen

src/abstractprobprog.jl

@@ -1,6 +1,7 @@
 using AbstractMCMC
 using DensityInterface
 using Random
+using StatsBase
 
 
 """
@@ -80,3 +81,29 @@ end
 function Base.rand(model::AbstractProbabilisticProgram)
     return rand(Random.default_rng(), NamedTuple, model)
 end
+
+"""
+    predict(
+        [rng::AbstractRNG=Random.default_rng(),]
+        [T=NamedTuple,]
+        model::AbstractProbabilisticProgram,
+        params,
+    ) -> T
+
+Draw a sample from the joint distribution specified by `model` conditioned on the values in
+`params`.
+
+The sample will be returned as format specified by `T`.
+"""
+function StatsBase.predict(rand::AbstractRNG, T::Type, model::AbstractProbabilisticProgram, params)
+    return rand(rng, T, condition(model, params))
+end
+function StatsBase.predict(T::Type, model::AbstractProbabilisticProgram, params)
+    return StatsBase.predict(Random.default_rng(), T, model, params)
+end
+function StatsBase.predict(model::AbstractProbabilisticProgram, params)
+    return StatsBase.predict(NamedTuple, model, params)
+end
+function StatsBase.predict(rng::AbstractRNG, params)
+    return StatsBase.predict(rng, NamedTuple, model, params)
+end