Stats bbo

SciML · Jan 4, 2024 · 94ed95b · 94ed95b
1 parent 34bff41
commit 94ed95b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/lib/OptimizationBBO/src/OptimizationBBO.jl b/lib/OptimizationBBO/src/OptimizationBBO.jl
@@ -178,7 +178,7 @@ function SciMLBase.__solve(cache::Optimization.OptimizationCache{
     t1 = time()
 
     opt_ret = Symbol(opt_res.stop_reason)
-    stats = Optimization.OptimizationStats(opt_res.iterations, t1 - t0, opt_res.f_calls)
+    stats = Optimization.OptimizationStats(; iterations = opt_res.iterations, time = t1 - t0, fevals = opt_res.f_calls)
     SciMLBase.build_solution(cache, cache.opt,
         BlackBoxOptim.best_candidate(opt_res),
         BlackBoxOptim.best_fitness(opt_res);

diff --git a/lib/OptimizationCMAEvolutionStrategy/src/OptimizationCMAEvolutionStrategy.jl b/lib/OptimizationCMAEvolutionStrategy/src/OptimizationCMAEvolutionStrategy.jl
@@ -105,11 +105,12 @@ function SciMLBase.__solve(cache::OptimizationCache{
     t1 = time()
 
     opt_ret = opt_res.stop.reason
-
+    stats = Optimization.OptimizationStats(; iterations = length(opt.logger.fmedian), time = t1 - t0, fevals = length(opt.logger.fmedian))
     SciMLBase.build_solution(cache, cache.opt,
         opt_res.logger.xbest[end],
         opt_res.logger.fbest[end]; original = opt_res,
-        retcode = opt_ret, solve_time = t1 - t0)
+        retcode = opt_ret, solve_time = t1 - t0,
+        stats = stats)
 end
 
 end