Downloads · 30 days
17
29% of all-time downloads
scibert_sdg_cased_zo-up
dvdblk/scibert_sdg_cased_zo-up
scibert_sdg_cased_zo-up is a text classification model from dvdblk. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
SciBERT fine-tuned on a custom dataset to predict the correct SDG class based on a scientific publication abstract.
All-time downloads
59
Public
Parameters
110M
1.3 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt880 MB · 67%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- bert
- Access
- Public
- Created
- Sep 4, 2024
- Updated
- Sep 4, 2024
- SHA
- 8614f3b0
Try a prompt
Base models
Model card
From the Hugging Face model README
SciBERT fine-tuned on SDG data
SciBERT fine-tuned on a custom dataset to predict the correct SDG class based on a scientific publication abstract.
Data
The training set consisted of 630 pairs of (abstract, sdg) with a balanced class distribution. The test set contains 271 samples.
Performance
| Metric | Value |
|---|---|
| Accuracy | 0.7528 |
| F1 | 0.7465 |
| Recall | 0.7528 |
| Precision | 0.7507 |
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- apache-2.0
- Languages
- en
- Created
- Sep 4, 2024
- Updated
- Sep 4, 2024