Downloads · 30 days
12
18% of all-time downloads
samirmsallem/gbert-base-text_parts
gbert-base-text_parts is a text classification model from samirmsallem. Use it when you need a label for a piece of text. It is set up for transformers.
gbert-base-textparts is a sequence classification model in the scientific domain in German, finetuned from the model gbert-base. It was trained using a custom annotated dataset of around 4,800 training and 1,400 test…
Downloads · 30 days
12
18% of all-time downloads
All-time downloads
66
Public
Parameters
110M
1.3 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt880 MB · 67%
From the Hugging Face model README
gbert-base-text_parts is a sequence classification model in the scientific domain in German, finetuned from the model gbert-base. It was trained using a custom annotated dataset of around 4,800 training and 1,400 test examples containing introduction, main part and conclusion text sequences from scientific papers and theses in german.
The two German BERT models — gbert-base and gbert-large — were evaluated on the same text classification task. This repository contains the model and results for gbert-base.
Despite its superior peak performance, gbert-large exhibited a consistent degradation in evaluation accuracy and an increase in loss over subsequent epochs. This pattern is characteristic of overfitting, where the model fits the training data too closely and fails to generalize well to unseen data. In contrast, gbert-base maintained more stable performance over time, making it a more robust choice in scenarios where computational resources or dataset size are limited.
| Text Classification Tag | Text Classification Label | Description |
|---|---|---|
| 0 | EINLEITUNG | The text is an introduction (Einleitung). |
| 1 | HAUPTTEIL | The text is a main part (Hauptteil). |
| 2 | SCHLUSS | The text is a conclusion (Schluss). |
Training was conducted on a 10 epoch fine-tuning approach:
| Epoch | Loss | Accuracy |
|---|---|---|
| 1.0 | 2.1272 | 0.6835 |
| 2.0 | 2.2270 | 0.7017 |
| 3.0 | 2.3057 | 0.7003 |
| 4.0 | 2.4637 | 0.6996 |
| 5.0 | 2.6276 | 0.6996 |
| 6.0 | 2.6948 | 0.7031 |
| 7.0 | 2.7076 | 0.7031 |
| 8.0 | 2.7590 | 0.7031 |
| 9.0 | 2.7891 | 0.7024 |
| 10.0 | 2.7976 | 0.7031 |
Training was conducted using a standard Text classification objective. The model achieves an accuracy of approximately 70% on the evaluation set.
Here are the overall final metrics on the test dataset after 10 epochs of training: