Código fuente de este video, en donde se muestra cómo re-entrenar a BERT para analizar sentimientos usando el set IMDB (movie reviews). La totalidad del código se puede ejecutar en Google Colab.
El set de entrenamiento se puede descargar de este enlace.