Hugging Face's logo

harr
/
distilbert-base-uncased-finetuned-ingredients

Token Classification
Transformers
PyTorch
TensorBoard
distilbert
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
2
harr commited on
Commit
a781a0b
·
1 Parent(s): 4e5658d

add model

Browse files
Files changed (7) hide show
  1. README.md +12 -12
  2. config.json +10 -10
  3. pytorch_model.bin +1 -1
  4. runs/Aug31_15-27-21_6005f014a2a9/1630423772.9026814/events.out.tfevents.1630423772.6005f014a2a9.74.1 +3 -0
  5. runs/Aug31_15-27-21_6005f014a2a9/events.out.tfevents.1630423772.6005f014a2a9.74.0 +3 -0
  6. runs/Aug31_15-27-21_6005f014a2a9/events.out.tfevents.1630423849.6005f014a2a9.74.2 +3 -0
  7. training_args.bin +1 -1
README.md CHANGED
@@ -22,16 +22,16 @@ model-index:
22
  metrics:
23
  - name: Precision
24
  type: precision
25
- value: 0.9878658101356174
26
  - name: Recall
27
  type: recall
28
- value: 0.9985569985569985
29
  - name: F1
30
  type: f1
31
- value: 0.9931826336562611
32
  - name: Accuracy
33
  type: accuracy
34
- value: 0.9928244463689224
35
  ---
36
 
37
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -41,11 +41,11 @@ should probably proofread and complete it, then remove this comment. -->
41
 
42
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the ingredients_yes_no dataset.
43
  It achieves the following results on the evaluation set:
44
- - Loss: 0.0280
45
- - Precision: 0.9879
46
- - Recall: 0.9986
47
- - F1: 0.9932
48
- - Accuracy: 0.9928
49
 
50
  ## Model description
51
 
@@ -76,9 +76,9 @@ The following hyperparameters were used during training:
76
 
77
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
78
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
79
- | No log | 1.0 | 186 | 0.0612 | 0.9807 | 0.9899 | 0.9853 | 0.9821 |
80
- | No log | 2.0 | 372 | 0.0390 | 0.9836 | 0.9935 | 0.9885 | 0.9890 |
81
- | 0.1179 | 3.0 | 558 | 0.0280 | 0.9879 | 0.9986 | 0.9932 | 0.9928 |
82
 
83
 
84
  ### Framework versions
 
22
  metrics:
23
  - name: Precision
24
  type: precision
25
+ value: 0.9658580413297394
26
  - name: Recall
27
  type: recall
28
+ value: 0.9649910233393177
29
  - name: F1
30
  type: f1
31
+ value: 0.9654243376740009
32
  - name: Accuracy
33
  type: accuracy
34
+ value: 0.9921109555923145
35
  ---
36
 
37
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
41
 
42
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the ingredients_yes_no dataset.
43
  It achieves the following results on the evaluation set:
44
+ - Loss: 0.0246
45
+ - Precision: 0.9659
46
+ - Recall: 0.9650
47
+ - F1: 0.9654
48
+ - Accuracy: 0.9921
49
 
50
  ## Model description
51
 
 
76
 
77
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
78
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
79
+ | No log | 1.0 | 184 | 0.0562 | 0.9177 | 0.9309 | 0.9242 | 0.9833 |
80
+ | No log | 2.0 | 368 | 0.0287 | 0.9605 | 0.9605 | 0.9605 | 0.9905 |
81
+ | 0.1154 | 3.0 | 552 | 0.0246 | 0.9659 | 0.9650 | 0.9654 | 0.9921 |
82
 
83
 
84
  ### Framework versions
config.json CHANGED
@@ -9,19 +9,19 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "ADD-B",
13
- "1": "ADD-C",
14
- "2": "REM-B",
15
- "3": "REM-C",
16
- "4": "O"
17
  },
18
  "initializer_range": 0.02,
19
  "label2id": {
20
- "ADD-B": 0,
21
- "ADD-C": 1,
22
- "O": 4,
23
- "REM-B": 2,
24
- "REM-C": 3
25
  },
26
  "max_position_embeddings": 512,
27
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "O",
13
+ "1": "B-ADD",
14
+ "2": "I-ADD",
15
+ "3": "B-REM",
16
+ "4": "I-REM"
17
  },
18
  "initializer_range": 0.02,
19
  "label2id": {
20
+ "B-ADD": 1,
21
+ "B-REM": 3,
22
+ "I-ADD": 2,
23
+ "I-REM": 4,
24
+ "O": 0
25
  },
26
  "max_position_embeddings": 512,
27
  "model_type": "distilbert",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:503cb238903261f0f8cac9dd517c723bb2ad363be5a5667f66e46e79222bb854
3
  size 265506293
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50e65301a3dda120b65ee0aea857bad12638b006c45a538114e0f66582066543
3
  size 265506293
runs/Aug31_15-27-21_6005f014a2a9/1630423772.9026814/events.out.tfevents.1630423772.6005f014a2a9.74.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d44bdbe38af1ee1acd40d50c75dad8b5eee222522f492b8262fec7ee3e40a12
3
+ size 4230
runs/Aug31_15-27-21_6005f014a2a9/events.out.tfevents.1630423772.6005f014a2a9.74.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fb6a60216adb6952f8a34c3093e9c99539bef925e15c105f860d24b1dd32e4f
3
+ size 5008
runs/Aug31_15-27-21_6005f014a2a9/events.out.tfevents.1630423849.6005f014a2a9.74.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ea0942ab11590e89e4b8b1efe1df45281624cbfcebee89998d9597448cef8db
3
+ size 512
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a682fa36f6065ff2cf83b4397ce8c7b995377546727d42eea2aa951aed9cdcb1
3
  size 2671
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:976675eaecd241738fd6e446f5dbf82cbb6ce8c4c1172b983771ac24721d67a5
3
  size 2671