It seems that I have been making some mistakes when I have been building my model, since I am getting the following error:
ValueError: Input contains NaN
