Skip to content

loretoparisi/bert_text_classifier

Repository files navigation

bert_text_classifier

Text Classification with BERT with Fastai and HuggingFace. Adapted from https://www.kaggle.com/maroberti/fastai-with-transformers-bert-roberta

How to run Notebooks

You can run BertTextClassifier.ipynb in a Jupyter Notebook or Google's Colab. To run within Microsoft VisualStudio Code use BertTextClassifier_VSCode.ipynb version.

How to run in command line

To run in command line please first install requirements in a virtual env:

python3 -m venv .env
pip3 install -r requirements.txt
python3 BertTextClassifier.py

Disclaimer

This project was freely adapted by Loreto Parisi from "Fastai with HuggingFace 🤗Transformers (BERT, RoBERTa, XLNet, XLM, DistilBERT)" - https://www.kaggle.com/maroberti/fastai-with-transformers-bert-roberta