Dear Comunity,

I am working on a prediction model using ANN. The script is written with the help of Pytorch. However, the prediction isn't really performed as desired (see the attached image). I have tried to tun the model by changing the learning rate, the neuron number, the hidden layer number, and even the data but no luck at all.

I would really appreciate if anyone can help fix my code or give me hints on how to improve the code (the code is attached as well as the data).

Thank you