Downloads · 30 days
8
4% of all-time downloads
autotrain-text-classific-imdb
neil-code/autotrain-text-classific-imdb
autotrain-text-classific-imdb is a text classification model from neil-code. Use it when you need a label for a piece of text. It is set up for transformers.
All-time downloads
228
Public
Parameters
109M
1.3 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt876 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jan 4, 2024
- Updated
- Jan 4, 2024
- SHA
- a4c16987
Try a prompt
Model card
From the Hugging Face model README
Model Trained Using AutoTrain
- Problem type: Text Classification
Validation Metrics
loss: 0.32178911566734314
f1: 0.8431746031746032
precision: 0.8952808988764045
recall: 0.7968
auc: 0.94403024
accuracy: 0.8518
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Created
- Jan 4, 2024
- Updated
- Jan 4, 2024