Skip to content

Commit

Permalink
docs: update docs for MLP and DeepSplitting
Browse files Browse the repository at this point in the history
  • Loading branch information
ashutosh-b-b committed Feb 9, 2024
1 parent 9ba1455 commit 125182d
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion docs/src/tutorials/deepsplitting.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ g(x) = exp.(- sum(x.^2, dims=1) ) # initial condition
σ(x, p, t) = 0.1f0 # diffusion coefficients
x0_sample = UniformSampling(fill(-5f-1, d), fill(5f-1, d))
f(x, y, v_x, v_y, ∇v_x, ∇v_y, p, t) = v_x .* (1f0 .- v_y)
prob = prob = PIDEProblem(μ, σ, x0, tspan, g, f; x0_sample = x0_sample)
prob = PIDEProblem(μ, σ, x0, tspan, g, f; x0_sample = x0_sample)

## Definition of the neural network to use
using Flux # needed to define the neural network
Expand Down
4 changes: 2 additions & 2 deletions docs/src/tutorials/mlp.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,8 @@ x0 = fill(0.,d) # initial point
g(x) = exp(- sum(x.^2) ) # initial condition
μ(x, p, t) = 0.0 # advection coefficients
σ(x, p, t) = 0.1 # diffusion coefficients
f(x, y, v_x, v_y, ∇v_x, ∇v_y, p, t) = max(0.0, v_x) * (1 - max(0.0, v_x)) # nonlocal nonlinear part of the
prob = PIDEProblem(μ, σ, x0, tspan, g, f) # defining the problem
f(x, v_x, ∇v_x, p, t) = max(0.0, v_x) * (1 - max(0.0, v_x)) # nonlocal nonlinear part of the
prob = ParabolicPDEProblem(μ, σ, x0, tspan, g, f) # defining the problem

## Definition of the algorithm
alg = MLP() # defining the algorithm. We use the Multi Level Picard algorithm
Expand Down

0 comments on commit 125182d

Please sign in to comment.