Skip to content

Commit d916b64

Browse files
committed
add easier model loading code
1 parent f5a2f9a commit d916b64

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

example/SyntheticTargets.jl

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
using Distributions, Random, LinearAlgebra
2+
using LogDensityProblems
3+
using IrrationalConstants
4+
using Plots
5+
6+
7+
include("targets/banana.jl")
8+
include("targets/cross.jl")
9+
include("targets/neal_funnel.jl")
10+
include("targets/warped_gaussian.jl")
11+
12+
13+
function load_model(name::String)
14+
if name == "Banana"
15+
return Banana(2, 1.0, 10.0)
16+
elseif name == "Cross"
17+
return Cross()
18+
elseif name == "Funnel"
19+
return Funnel(2)
20+
elseif name == "WarpedGaussian"
21+
return WarpedGauss()
22+
else
23+
error("Model not defined")
24+
end
25+
end
26+
27+
LogDensityProblems.dimension(dist::ContinuousDistribution) = length(dist)
28+
LogDensityProblems.logdensity(dist::ContinuousDistribution, x) = logpdf(dist, x)
29+

0 commit comments

Comments
 (0)