nkzhlee / Data_Augmentation

Text-Classification-with-BERT-PyTorch-Implementation

A text classifier by fine-tuning on pre-trained Huggingface models

Dataset

TREC

Pre Requisite

• python 3.7
• pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
• pip install numpy spacy tqdm wandb
• pip install transformers[pytorch]

Results (Train Accuracy / Validation Accuracy)