From 51ecb372db953bc329f995e0e9f65016cb0192fb Mon Sep 17 00:00:00 2001 From: Astitva Aggarwal Date: Fri, 1 Sep 2023 13:47:33 +0530 Subject: [PATCH] minor changes --- src/advancedHMC_MCMC.jl | 13 +++++++------ test/BPINN_Tests.jl | 7 +++---- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/src/advancedHMC_MCMC.jl b/src/advancedHMC_MCMC.jl index 2483a16af..533366923 100644 --- a/src/advancedHMC_MCMC.jl +++ b/src/advancedHMC_MCMC.jl @@ -195,15 +195,16 @@ end # My suggested extra loss function function L2loss2(Tar::LogTargetDensity, θ) f = Tar.prob.f - dataset = Tar.dataset - - # Timepoints to enforce Physics - dataset = Array(reduce(hcat, dataset)') - t = dataset[end, :] - û = dataset[1:(end - 1), :] # parameter estimation chosen or not if Tar.extraparams > 0 + dataset = Tar.dataset + + # Timepoints to enforce Physics + dataset = Array(reduce(hcat, dataset)') + t = dataset[end, :] + û = dataset[1:(end - 1), :] + ode_params = Tar.extraparams == 1 ? θ[((length(θ) - Tar.extraparams) + 1):length(θ)][1] : θ[((length(θ) - Tar.extraparams) + 1):length(θ)] diff --git a/test/BPINN_Tests.jl b/test/BPINN_Tests.jl index cd9b2093d..144b63bb0 100644 --- a/test/BPINN_Tests.jl +++ b/test/BPINN_Tests.jl @@ -130,10 +130,9 @@ fh_mcmc_chain1, fhsamples1, fhstats1 = ahmc_bayesian_pinn_ode(prob, chainflux1, 3.0), param = [ LogNormal(9, - 0.5), + 5), ], - Metric = DiagEuclideanMetric, - n_leapfrog = 30) + Metric = DiagEuclideanMetric) fh_mcmc_chain2, fhsamples2, fhstats2 = ahmc_bayesian_pinn_ode(prob, chainlux1, dataset = dataset, @@ -187,7 +186,7 @@ meanscurve2 = prob.u0 .+ (t .- prob.tspan[1]) .* luxmean # ESTIMATED ODE PARAMETERS (NN1 AND NN2) @test abs(p - mean([fhsamples2[i][23] for i in 2000:2500])) < abs(0.2 * p) -@test abs(p - mean([fhsamples1[i][26] for i in 2000:2500])) < abs(0.2 * p) +@test abs(p - mean([fhsamples1[i][23] for i in 2000:2500])) < abs(0.2 * p) #---------------------- solve() call @test mean(abs.(x̂1 .- sol2flux.ensemblesol)) < 5e-1