Couldn’t train nn for solving 2nd order ODE

I am trying to solve 2nd order ODE which is

y”+100y=0, y(0)=0, y'(0)=10 on [0, 1] interval

using neural network. Here is the code:

But I have two problems:

  1. I guess tf recompiles(retraces) some function during learning which slows learning proccess significantly. Putting whole learning process into function doesn’t help.
  2. NN doesn’t fit at all. I guess it might be because of gradient size on last layer or something. Anyway it is difficult to test during 1.

Any help with problem 1 and maybe problem 2 will be appreciated.

submitted by /u/rprtr258
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *