Skip to content

Commit c77dc2d

Browse files
committed
rm load_model and use constructor directly
1 parent 427225e commit c77dc2d

File tree

3 files changed

+2
-17
lines changed

3 files changed

+2
-17
lines changed

example/SyntheticTargets.jl

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -9,21 +9,6 @@ include("targets/cross.jl")
99
include("targets/neal_funnel.jl")
1010
include("targets/warped_gaussian.jl")
1111

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-
2712
function visualize(p::ContinuousMultivariateDistribution, samples=rand(p, 1000))
2813
xrange = range(minimum(samples[1, :]) - 1, maximum(samples[1, :]) + 1; length=100)
2914
yrange = range(minimum(samples[2, :]) - 1, maximum(samples[2, :]) + 1; length=100)

example/demo_planar_flow.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ T = Float64
1313
######################################
1414
# 2d Banana as the target distribution
1515
######################################
16-
target = load_model("Banana")
16+
target = Banana(2, 1.0, 10.0)
1717
logp = Base.Fix1(logpdf, target)
1818

1919

example/demo_radial_flow.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ T = Float64
1313
######################################
1414
# get target logp
1515
######################################
16-
target = load_model("WarpedGaussian")
16+
target = WarpedGauss()
1717
logp = Base.Fix1(logpdf, target)
1818

1919
######################################

0 commit comments

Comments
 (0)