I am trying to train a simple Neural Network and I got this error, I tried some other answers I find in similar questions and it did not work. I want to classify between TYP