En este repo se ve como armar un custom data generator para entrenar con Keras
La data se puede bajar de la competencia de kaggle: https://www.kaggle.com/competitions/jigsaw-toxic-comment-classification-challenge/data
conda create -n toxic-comments python=3.12
conda activate toxic-comments
pip install -f requirements.txt