VikasSingh-DS/Movies-Reviews-Bert-Sentiment-Flask-API

Fine-tune BERT for sentiment analysis. I have done text preprocessing (special tokens, padding, and attention masks) and build a Sentiment Classifier using the amazing Transformers library by Hugging Face! I have train my model on kaggle notebook on gpu. The model give the accuracy of 95.14% on validation dataset.

语言：Jupyter Notebook · GitHub 星标：9 · Fork：0

话题：bert-model · flask · gpu · huggingface · pytorch · sentiment-analysis

在 GitHub 上查看原仓库