@@ -2,23 +2,23 @@ using OptimizationBase
2
2
using OptimizationBase: ReturnCode
3
3
using OptimizationBase. SciMLBase: OptimizationFunction, OptimizationProblem
4
4
using ForwardDiff, Zygote
5
- using OptimizationLBFGS
5
+ using OptimizationLBFGSB
6
6
using MLUtils
7
7
using LBFGSB
8
8
using Test
9
9
10
- @testset " OptimizationLBFGS .jl" begin
10
+ @testset " OptimizationLBFGSB .jl" begin
11
11
x0 = zeros (2 )
12
12
rosenbrock (x, p = nothing ) = (1 - x[1 ])^ 2 + 100 * (x[2 ] - x[1 ]^ 2 )^ 2
13
13
l1 = rosenbrock (x0)
14
14
15
15
optf = OptimizationFunction (rosenbrock, OptimizationBase. AutoForwardDiff ())
16
16
prob = OptimizationProblem (optf, x0)
17
- @time res = solve (prob, OptimizationLBFGS . LBFGS (), maxiters = 100 )
17
+ @time res = solve (prob, OptimizationLBFGSB . LBFGS (), maxiters = 100 )
18
18
@test res. retcode == ReturnCode. Success
19
19
20
20
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 )
22
22
@test res. retcode == ReturnCode. Success
23
23
24
24
function con2_c (res, x, p)
@@ -29,7 +29,7 @@ using Test
29
29
prob = OptimizationProblem (optf, x0, lcons = [1.0 , - Inf ],
30
30
ucons = [1.0 , 0.0 ], lb = [- 1.0 , - 1.0 ],
31
31
ub = [1.0 , 1.0 ])
32
- @time res = solve (prob, OptimizationLBFGS . LBFGS (), maxiters = 100 )
32
+ @time res = solve (prob, OptimizationLBFGSB . LBFGS (), maxiters = 100 )
33
33
@test res. retcode == SciMLBase. ReturnCode. Success
34
34
35
35
x0 = (- pi ): 0.001 : pi
@@ -52,6 +52,6 @@ using Test
52
52
l0 = optf (initpars, (x0, y0))
53
53
prob = OptimizationProblem (optf, initpars, (x0, y0), lcons = [- Inf ], ucons = [0.5 ],
54
54
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)
56
56
@test opt1. objective < l0
57
57
end
0 commit comments