I\'m working through some exercises in Tensorflow 2.0 to make sure I understand the basics properly. My current task is to fit a regression with an MSE function using keras