Skip to content

Commit 93bcef3

Browse files
committed
bench: new MHE benchmarks with hessian=true
1 parent 46b47be commit 93bcef3

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

benchmark/2_bench_state_estim.jl

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -305,6 +305,13 @@ mhe_pendulum_ipopt_curr = setconstraint!(mhe_pendulum_ipopt_curr; v̂min, v̂max
305305
JuMP.unset_time_limit_sec(mhe_pendulum_ipopt_curr.optim)
306306
JuMP.set_attribute(mhe_pendulum_ipopt_curr.optim, "tol", 1e-7)
307307

308+
hessian = true
309+
mhe_pendulum_ipopt_currh = MovingHorizonEstimator(
310+
model; He, σQ, σR, nint_u, σQint_u, optim, direct, hessian
311+
)
312+
mhe_pendulum_ipopt_currh = setconstraint!(mhe_pendulum_ipopt_currh; v̂min, v̂max)
313+
JuMP.unset_time_limit_sec(mhe_pendulum_ipopt_currh.optim)
314+
308315
optim = JuMP.Model(optimizer_with_attributes(Ipopt.Optimizer,"sb"=>"yes"), add_bridges=false)
309316
direct = false
310317
mhe_pendulum_ipopt_pred = MovingHorizonEstimator(
@@ -314,6 +321,14 @@ mhe_pendulum_ipopt_pred = setconstraint!(mhe_pendulum_ipopt_pred; v̂min, v̂max
314321
JuMP.unset_time_limit_sec(mhe_pendulum_ipopt_pred.optim)
315322
JuMP.set_attribute(mhe_pendulum_ipopt_pred.optim, "tol", 1e-7)
316323

324+
hessian = true
325+
mhe_pendulum_ipopt_predh = MovingHorizonEstimator(
326+
model; He, σQ, σR, nint_u, σQint_u, optim, direct, hessian
327+
)
328+
mhe_pendulum_ipopt_predh = setconstraint!(mhe_pendulum_ipopt_predh; v̂min, v̂max)
329+
JuMP.unset_time_limit_sec(mhe_pendulum_ipopt_predh.optim)
330+
JuMP.set_attribute(mhe_pendulum_ipopt_predh.optim, "tol", 1e-7)
331+
317332
optim = JuMP.Model(MadNLP.Optimizer, add_bridges=false)
318333
direct = true
319334
mhe_pendulum_madnlp_curr = MovingHorizonEstimator(
@@ -338,11 +353,21 @@ CASE_ESTIM["Pendulum"]["MovingHorizonEstimator"]["Ipopt"]["Current form"] =
338353
sim!($mhe_pendulum_ipopt_curr, $N, $u; plant=$plant, x_0=$x_0, x̂_0=$x̂_0, progress=false),
339354
samples=samples, evals=evals, seconds=seconds
340355
)
356+
CASE_ESTIM["Pendulum"]["MovingHorizonEstimator"]["Ipopt"]["Current form (Hessian)"] =
357+
@benchmarkable(
358+
sim!($mhe_pendulum_ipopt_currh, $N, $u; plant=$plant, x_0=$x_0, x̂_0=$x̂_0, progress=false),
359+
samples=samples, evals=evals, seconds=seconds
360+
)
341361
CASE_ESTIM["Pendulum"]["MovingHorizonEstimator"]["Ipopt"]["Prediction form"] =
342362
@benchmarkable(
343363
sim!($mhe_pendulum_ipopt_pred, $N, $u; plant=$plant, x_0=$x_0, x̂_0=$x̂_0, progress=false),
344364
samples=samples, evals=evals, seconds=seconds
345365
)
366+
CASE_ESTIM["Pendulum"]["MovingHorizonEstimator"]["Ipopt"]["Prediction form (Hessian)"] =
367+
@benchmarkable(
368+
sim!($mhe_pendulum_ipopt_predh, $N, $u; plant=$plant, x_0=$x_0, x̂_0=$x̂_0, progress=false),
369+
samples=samples, evals=evals, seconds=seconds
370+
)
346371
CASE_ESTIM["Pendulum"]["MovingHorizonEstimator"]["MadNLP"]["Current form"] =
347372
@benchmarkable(
348373
sim!($mhe_pendulum_madnlp_curr, $N, $u; plant=$plant, x_0=$x_0, x̂_0=$x̂_0, progress=false),

0 commit comments

Comments
 (0)