Update README.md
Browse files
README.md
CHANGED
@@ -10,4 +10,38 @@ tags:
|
|
10 |
- nlp
|
11 |
- bert
|
12 |
- fine-tune
|
13 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
- nlp
|
11 |
- bert
|
12 |
- fine-tune
|
13 |
+
---
|
14 |
+
|
15 |
+
|
16 |
+
# distilbert-base-uncased-finetuned-fashion
|
17 |
+
|
18 |
+
This model is a fine-tuned version of [bert-base-french-europeana-cased](https://huggingface.co/dbmdz/bert-base-french-europeana-cased) on a munally created dataset in order to detect fashion (label_0) from non-fashion (label_1) items.
|
19 |
+
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 1.21
|
21 |
+
- Accuracy: 0.85
|
22 |
+
|
23 |
+
### Training hyperparameters
|
24 |
+
|
25 |
+
The following hyperparameters were used during training:
|
26 |
+
- learning_rate: 2e-05
|
27 |
+
- train_batch_size: 64
|
28 |
+
- eval_batch_size: 64
|
29 |
+
- seed: 42
|
30 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
31 |
+
- num_epochs: 10
|
32 |
+
|
33 |
+
### Training results
|
34 |
+
|
35 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy
|
36 |
+
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
37 |
+
| 4.2 | 1.0 | 47 | 4.15156 | 0.174 |
|
38 |
+
...
|
39 |
+
| 1.216 | 10 | 490 | 1.2586 | 0.856 |
|
40 |
+
|
41 |
+
|
42 |
+
### Framework versions
|
43 |
+
|
44 |
+
- Transformers 4.18.0
|
45 |
+
- Pytorch 1.11.0+cu113
|
46 |
+
- Datasets 2.1.0
|
47 |
+
- Tokenizers 0.12.1
|