- The whole experiment environment is tested and built on Google Colab.
- This classifier is built with Transformer based models, LSTM and other NLP classifiers.
- This project is allowed to run on Google Colab with single GPU.
If you prefer to work on local environment, this repo is able to run using Jupyter Notebook. - Since BERT model is developed under
Tensorflow 1.x
, more modification to get compatible withTensorflow 2.x
is already added to codes.
To run the classifier, follow the instructions below...
- If running on local environment, import BERT_Demo to Jupyter Notebook.
- If running on Google Colab, import BERT_Demo directly.