README.md

Sentiment Classfication on IMDB Dataset

Supervised sentiment classification using fine-tuned BERT with PyTorch

Transformers package is from huggingface

Contents

• create_token.py
• prepare_for_bert.py
• prepre_for_bert_result.py
• dataloader.py
• load_bert.py
• fine_tune_bert.py