Build Text Classification Models (LSTM, BERT, etc.) for English data


Codes for building text classification models for English data

Published by Haerim Hwang

language model LSTM BERT text classification natural language processing google colab python

0 min READ

  • This script builds a couple of machine learning models to classify documents by pre-set categories. We use Google Colab for the script, which allows us to use GPU instead of CPU.


  • Click HERE for the Python script in Google Colab.

  • These codes were revised based on the following references.

    https://colab.research.google.com/github/shranith/ML-notebooks/blob/master/text_classification_20ng.ipynb

    https://www.kaggle.com/dikshabhati2002/text-classification-lstm-bert-ml-models