I organized the code and added comments to improve its readability.
Also, the current set up of the neural network makes it very unlikely to converge. To fix this, I increased the learning rate, changed the kernel initialization, and changed the output activation to linear since it is a regression problem.
I organized the code and added comments to improve its readability.
Also, the current set up of the neural network makes it very unlikely to converge. To fix this, I increased the learning rate, changed the kernel initialization, and changed the output activation to linear since it is a regression problem.