Downloads · 30 days
0
Junyi-BERT-sentiment-analysis-IMBD
Junyi-Shen/Junyi-BERT-sentiment-analysis-IMBD
Junyi-BERT-sentiment-analysis-IMBD is a text classification model from Junyi-Shen. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
Binary sentiment classification (positive / negative)
Access
Public
Updated Apr 13, 2026
Repo size
1.7 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt1.2 GB · 74%
At a glance
- Task
- Text Classification
- License
- apache-2.0
- Access
- Public
- Created
- Dec 15, 2025
- Updated
- Apr 13, 2026
- SHA
- 857e157a
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
Task
Binary sentiment classification (positive / negative)
Source Code
Training Data
-
Stage 1: Large-scale sentiment pre-finetuning
- Dataset Source: IMDB Dataset of 50K Movie Reviews
-
Stage 2: IMDB sentiment fine-tuning
- Dataset Source: IMDB sentiment (https://github.com/acmi-lab/counterfactually-augmented-data/blob/master/sentiment/new/train.tsv)
Training Method
- Loss function: Binary Cross-Entropy (BCE)
- Weight decay: 0
- Batch size: 8
- Dropout: 0.3
- Learning rate: 2e-5
- Learning rate scheduler: Enabled
- Frozen layers: First 9 transformer layers
Evaluation
- Test dataset: IMDB sentiment (https://github.com/acmi-lab/counterfactually-augmented-data/blob/master/sentiment/new/test.tsv)
- Test accuracy: 98.4%
- Task
- Text Classification
- License
- apache-2.0
- Languages
- en
- Created
- Dec 15, 2025
- Updated
- Apr 13, 2026