As exercise I tried to follow and the copy the ungraded lab 5 of c3w3. I tried using a my own data set of anime synopsis and its review scores from 2-10. I adjusted the last layer of the model to make it a softmax activation with 9 neurons. When I tried training the model, the accuracy was very low and became 0 and the loss just outputted nan. Attached is my notebook and the json file of the dataset I used in a zip file. Could someone check out my code and see what I am doing wrong?
nlp exercise.zip (2.3 MB)