rasta commited on
Commit
8cd5d90
1 Parent(s): 3c55e29

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +35 -1
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