@@ -34,8 +34,8 @@ prob_dde = DDE.DDEProblem(delay_lotka_volterra!, u0, h, (0.0, 10.0),
3434 constant_lags = [0.1])
3535
3636function predict_dde(p)
37- return Array(ODE.solve(prob_dde, DDE.MethodOfSteps(ODE.Tsit5()),
38- u0 = u0, p = p, saveat = 0.1, sensealg = SMS.ReverseDiffAdjoint()))
37+ return Array(ODE.solve(prob_dde, DDE.MethodOfSteps(ODE.Tsit5());
38+ u0, p, saveat = 0.1, sensealg = SMS.ReverseDiffAdjoint()))
3939end
4040
4141loss_dde(p) = sum(abs2, x - 1 for x in predict_dde(p))
5353adtype = OPT.AutoZygote()
5454optf = OPT.OptimizationFunction((x, p) -> loss_dde(x), adtype)
5555optprob = OPT.OptimizationProblem(optf, p)
56- result_dde = OPT.solve(optprob, OPA.PolyOpt(), maxiters = 300, callback = callback)
56+ result_dde = OPT.solve(optprob, OPA.PolyOpt(); maxiters = 300, callback)
5757```
5858
5959Notice that we chose ` sensealg = ReverseDiffAdjoint() ` to utilize the ReverseDiff.jl
@@ -79,5 +79,5 @@ We use `Optimization.solve` to optimize the parameters for our loss function:
7979adtype = OPT.AutoZygote()
8080optf = OPT.OptimizationFunction((x, p) -> loss_dde(x), adtype)
8181optprob = OPT.OptimizationProblem(optf, p)
82- result_dde = OPT.solve(optprob, OPA.PolyOpt(), callback = callback)
82+ result_dde = OPT.solve(optprob, OPA.PolyOpt(); callback)
8383```
0 commit comments