Closed DesmondYuan closed 3 years ago
A quick workaround is passing additional arguments to the prediction to disable the saveat
, instead of output all of the inner datapoints.
https://github.com/jiweiqi/CellBox.jl/blob/main/spilt/network.jl#L65
Change the plotting functions as well.
Something like
function predict_neuralode(u0, p, i_exp=1, batch=ntotal; dense=false)
global μ = μ_list[i_exp, 1:ns]
_prob = remake(prob, p=p, tspan=[0, ts[batch]])
pred = Array(solve(_prob, Tsit5(), saveat=ts[1:batch], sensealg=InterpolatingAdjoint()))
return pred
end
Sounds perfect. Let me implement it and test
Implemented in 4a0711b301e5410d8d7a8e5793040d0706b739c1
nplot
for num of points when plot solutionsaveat
as an option in function predict_neuralode()
What should be the best way to fix this? @jiweiqi