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 |
|