https://learnmodernpython.com/tensorflow-building-a-simple-text-classification-model/