Update README.md
Browse files
README.md
CHANGED
|
@@ -22,27 +22,27 @@ This repository contains a model trained to predict Common European Framework of
|
|
| 22 |
- Model architecture: BERT (base model: `bert-base-...`)
|
| 23 |
- Task: CEFR level prediction for text classification
|
| 24 |
- Training dataset: CEFR dataset
|
| 25 |
-
- Fine-tuning: Epochs, Loss,
|
| 26 |
|
| 27 |
## Performance
|
| 28 |
|
| 29 |
The model's performance during training is summarized below:
|
| 30 |
|
| 31 |
|
| 32 |
-
| Epoch | Training Loss | Validation Loss
|
| 33 |
-
|-------|---------------|-----------------
|
| 34 |
-
| 1 | 0.
|
| 35 |
-
| 2 | 0.
|
| 36 |
-
| 3 | 0.
|
| 37 |
-
| 4 | 0.
|
| 38 |
-
| 5 | 0.
|
| 39 |
|
| 40 |
-
Additional metrics:
|
| 41 |
|
| 42 |
-
-
|
| 43 |
-
-
|
| 44 |
-
-
|
| 45 |
-
-
|
| 46 |
|
| 47 |
## Usage
|
| 48 |
|
|
|
|
| 22 |
- Model architecture: BERT (base model: `bert-base-...`)
|
| 23 |
- Task: CEFR level prediction for text classification
|
| 24 |
- Training dataset: CEFR dataset
|
| 25 |
+
- Fine-tuning: Epochs, Loss, etc.
|
| 26 |
|
| 27 |
## Performance
|
| 28 |
|
| 29 |
The model's performance during training is summarized below:
|
| 30 |
|
| 31 |
|
| 32 |
+
| Epoch | Training Loss | Validation Loss |
|
| 33 |
+
|-------|---------------|-----------------|
|
| 34 |
+
| 1 | 0.412300 | 0.396337 |
|
| 35 |
+
| 2 | 0.369600 | 0.388866 |
|
| 36 |
+
| 3 | 0.298200 | 0.419018 |
|
| 37 |
+
| 4 | 0.214500 | 0.481886 |
|
| 38 |
+
| 5 | 0.148300 | 0.557343 |
|
| 39 |
|
| 40 |
+
--Additional metrics:
|
| 41 |
|
| 42 |
+
--Training Loss: 0.2900624789151278
|
| 43 |
+
--Training Runtime: 5168.3962 seconds
|
| 44 |
+
--Training Samples per Second: 10.642
|
| 45 |
+
--Total Floating Point Operations: 1.447162776576e+16
|
| 46 |
|
| 47 |
## Usage
|
| 48 |
|