Skip to content

Commit

Permalink
Format .jl files (#391)
Browse files Browse the repository at this point in the history
Co-authored-by: prbzrg <[email protected]>
  • Loading branch information
github-actions[bot] and prbzrg authored Mar 27, 2024
1 parent 5a47046 commit abb62ab
Show file tree
Hide file tree
Showing 6 changed files with 668 additions and 668 deletions.
42 changes: 21 additions & 21 deletions src/cores/core.jl
Original file line number Diff line number Diff line change
@@ -1,21 +1,21 @@
# SciML interface

function callback_f(ps, l, ::AbstractICNF, prgr::Progress, itr_n::AbstractArray)
ProgressMeter.next!(
prgr;
showvalues = [
(:loss_value, l),
(:iteration, only(itr_n)),
(:last_update, Dates.now()),
],
)
itr_n[] += one(only(itr_n))
false
end

# MLJ interface

function MLJModelInterface.fitted_params(::MLJICNF, fitresult)
(ps, st) = fitresult
(learned_parameters = ps, states = st)
end
# SciML interface

function callback_f(ps, l, ::AbstractICNF, prgr::Progress, itr_n::AbstractArray)
ProgressMeter.next!(
prgr;
showvalues = [
(:loss_value, l),
(:iteration, only(itr_n)),
(:last_update, Dates.now()),
],
)
itr_n[] += one(only(itr_n))
false
end

# MLJ interface

function MLJModelInterface.fitted_params(::MLJICNF, fitresult)
(ps, st) = fitresult
(learned_parameters = ps, states = st)
end
Loading

0 comments on commit abb62ab

Please sign in to comment.