Skip to content

Commit

Permalink
Update docs/src/tutorials/pino_ode.md
Browse files Browse the repository at this point in the history
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
  • Loading branch information
KirillZubov and github-actions[bot] authored Sep 2, 2024
1 parent d0665ec commit 5295ceb
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions docs/src/tutorials/pino_ode.md
Original file line number Diff line number Diff line change
Expand Up @@ -55,9 +55,10 @@ function get_trainset(bounds, tspan, number_of_parameters, dt)
end
# Compute the ground truth solution for each parameter
ground_solution = (u0, p, t) -> u0 + p[1] / p[2] * sin(p[2] * t) + p[3]*t
function ground_solution_f(p,t)
reduce(hcat,[[ground_solution(u0, p[:, i], t[j]) for j in axes(t, 2)] for i in axes(p, 2)])
ground_solution = (u0, p, t) -> u0 + p[1] / p[2] * sin(p[2] * t) + p[3] * t
function ground_solution_f(p, t)
reduce(hcat,
[[ground_solution(u0, p[:, i], t[j]) for j in axes(t, 2)] for i in axes(p, 2)])
end
# generate the solution with new parameters for test the model
Expand Down

0 comments on commit 5295ceb

Please sign in to comment.