Hi,
I've created a Google Colab version of the code we worked on today.
If you want to give it a try and run it directly on the cloud for free please go to
this link. It already has all the dependencies installed, so you do not need anything but a Google account.
With this code I got into this:

I'm sure that this may help others to achieve better results.
Anyway, I just wanted to mention that I only used one hidden layer, thinking on the fact that we could approximate any function with only or hidden layer.
Best,
Rodolfo F.