https://gpttutorpro.com/pytorch-for-nlp-building-a-text-classifier/