BrandonM001
commited on
Commit
•
ab5f083
1
Parent(s):
d3ef9e6
Training complete
Browse files
README.md
CHANGED
@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
|
|
20 |
|
21 |
This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
|
22 |
It achieves the following results on the evaluation set:
|
23 |
-
- Loss:
|
24 |
-
- Precision: 0.
|
25 |
-
- Recall: 0.
|
26 |
-
- F1: 0.
|
27 |
-
- Accuracy: 0.
|
28 |
|
29 |
## Model description
|
30 |
|
@@ -43,23 +43,21 @@ More information needed
|
|
43 |
### Training hyperparameters
|
44 |
|
45 |
The following hyperparameters were used during training:
|
46 |
-
- learning_rate:
|
47 |
- train_batch_size: 8
|
48 |
- eval_batch_size: 8
|
49 |
- seed: 42
|
50 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
51 |
- lr_scheduler_type: linear
|
52 |
-
- num_epochs:
|
53 |
|
54 |
### Training results
|
55 |
|
56 |
-
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1
|
57 |
-
|
58 |
-
|
|
59 |
-
|
|
60 |
-
|
|
61 |
-
| 1.0241 | 4.0 | 7024 | 1.1057 | 0.0 | 0.0 | 0.0 | 0.7612 |
|
62 |
-
| 1.0372 | 5.0 | 8780 | 1.1037 | 0.0 | 0.0 | 0.0 | 0.7612 |
|
63 |
|
64 |
|
65 |
### Framework versions
|
|
|
20 |
|
21 |
This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
|
22 |
It achieves the following results on the evaluation set:
|
23 |
+
- Loss: 0.0638
|
24 |
+
- Precision: 0.9377
|
25 |
+
- Recall: 0.9500
|
26 |
+
- F1: 0.9438
|
27 |
+
- Accuracy: 0.9858
|
28 |
|
29 |
## Model description
|
30 |
|
|
|
43 |
### Training hyperparameters
|
44 |
|
45 |
The following hyperparameters were used during training:
|
46 |
+
- learning_rate: 2e-05
|
47 |
- train_batch_size: 8
|
48 |
- eval_batch_size: 8
|
49 |
- seed: 42
|
50 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
51 |
- lr_scheduler_type: linear
|
52 |
+
- num_epochs: 3
|
53 |
|
54 |
### Training results
|
55 |
|
56 |
+
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|
57 |
+
|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
|
58 |
+
| 0.0737 | 1.0 | 1756 | 0.0652 | 0.9077 | 0.9335 | 0.9204 | 0.9817 |
|
59 |
+
| 0.0358 | 2.0 | 3512 | 0.0679 | 0.9307 | 0.9450 | 0.9378 | 0.9850 |
|
60 |
+
| 0.0209 | 3.0 | 5268 | 0.0638 | 0.9377 | 0.9500 | 0.9438 | 0.9858 |
|
|
|
|
|
61 |
|
62 |
|
63 |
### Framework versions
|