https://sourcedexter.com/tensorflow-text-classification-python/