https://machinelearningmodels.org/building-a-custom-text-classification-model-using-python/