@@ -2,23 +2,23 @@ using OptimizationBase
22using OptimizationBase: ReturnCode
33using OptimizationBase. SciMLBase: OptimizationFunction, OptimizationProblem
44using ForwardDiff, Zygote
5- using OptimizationLBFGS
5+ using OptimizationLBFGSB
66using MLUtils
77using LBFGSB
88using Test
99
10- @testset " OptimizationLBFGS .jl" begin
10+ @testset " OptimizationLBFGSB .jl" begin
1111 x0 = zeros (2 )
1212 rosenbrock (x, p = nothing ) = (1 - x[1 ])^ 2 + 100 * (x[2 ] - x[1 ]^ 2 )^ 2
1313 l1 = rosenbrock (x0)
1414
1515 optf = OptimizationFunction (rosenbrock, OptimizationBase. AutoForwardDiff ())
1616 prob = OptimizationProblem (optf, x0)
17- @time res = solve (prob, OptimizationLBFGS . LBFGS (), maxiters = 100 )
17+ @time res = solve (prob, OptimizationLBFGSB . LBFGS (), maxiters = 100 )
1818 @test res. retcode == ReturnCode. Success
1919
2020 prob = OptimizationProblem (optf, x0, lb = [- 1.0 , - 1.0 ], ub = [1.0 , 1.0 ])
21- @time res = solve (prob, OptimizationLBFGS . LBFGS (), maxiters = 100 )
21+ @time res = solve (prob, OptimizationLBFGSB . LBFGS (), maxiters = 100 )
2222 @test res. retcode == ReturnCode. Success
2323
2424 function con2_c (res, x, p)
@@ -29,7 +29,7 @@ using Test
2929 prob = OptimizationProblem (optf, x0, lcons = [1.0 , - Inf ],
3030 ucons = [1.0 , 0.0 ], lb = [- 1.0 , - 1.0 ],
3131 ub = [1.0 , 1.0 ])
32- @time res = solve (prob, OptimizationLBFGS . LBFGS (), maxiters = 100 )
32+ @time res = solve (prob, OptimizationLBFGSB . LBFGS (), maxiters = 100 )
3333 @test res. retcode == SciMLBase. ReturnCode. Success
3434
3535 x0 = (- pi ): 0.001 : pi
@@ -52,6 +52,6 @@ using Test
5252 l0 = optf (initpars, (x0, y0))
5353 prob = OptimizationProblem (optf, initpars, (x0, y0), lcons = [- Inf ], ucons = [0.5 ],
5454 lb = [- 10.0 , - 10.0 , - 10.0 , - 10.0 , - 10.0 ], ub = [10.0 , 10.0 , 10.0 , 10.0 , 10.0 ])
55- opt1 = solve (prob, OptimizationLBFGS . LBFGS (), maxiters = 1000 , callback = callback)
55+ opt1 = solve (prob, OptimizationLBFGSB . LBFGS (), maxiters = 1000 , callback = callback)
5656 @test opt1. objective < l0
5757end
0 commit comments