Thank you! Since I was focusing on the implementation I completely missed that the output is being clipped to a positive number. I’ll increase the number of epochs and will also try with different piecewise linear models to get a better intuition of how many parameters to use, I’ll probably end up with a fairly wide network with a couple of layers lol. Thanks again!