Hugging Face's logo

Saint5
/
hf_tutorial_food_not_food_text_classifier-distilbert-base-uncased

Text Classification
Transformers
Safetensors
distilbert
Generated from Trainer
Model card Files
xet
Community
Saint5 commited on
Commit
2b32bb2
·
verified ·
1 Parent(s): 86a85d9

Second commit of the text classifier model

Browse files
Files changed (4) hide show
  1. README.md +12 -12
  2. model.safetensors +1 -1
  3. tokenizer.json +14 -2
  4. training_args.bin +1 -1
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0004
22
- - Accuracy: 1.0
23
 
24
  ## Model description
25
 
@@ -50,16 +50,16 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
- | 0.3606 | 1.0 | 7 | 0.0487 | 1.0 |
54
- | 0.024 | 2.0 | 14 | 0.0057 | 1.0 |
55
- | 0.0039 | 3.0 | 21 | 0.0019 | 1.0 |
56
- | 0.0016 | 4.0 | 28 | 0.0010 | 1.0 |
57
- | 0.001 | 5.0 | 35 | 0.0007 | 1.0 |
58
- | 0.0008 | 6.0 | 42 | 0.0006 | 1.0 |
59
- | 0.0007 | 7.0 | 49 | 0.0005 | 1.0 |
60
- | 0.0006 | 8.0 | 56 | 0.0005 | 1.0 |
61
- | 0.0006 | 9.0 | 63 | 0.0004 | 1.0 |
62
- | 0.0006 | 10.0 | 70 | 0.0004 | 1.0 |
63
 
64
 
65
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0687
22
+ - Accuracy: 0.98
23
 
24
  ## Model description
25
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 0.4267 | 1.0 | 7 | 0.1057 | 0.98 |
54
+ | 0.0479 | 2.0 | 14 | 0.0092 | 1.0 |
55
+ | 0.0053 | 3.0 | 21 | 0.0480 | 0.98 |
56
+ | 0.0019 | 4.0 | 28 | 0.0694 | 0.98 |
57
+ | 0.0011 | 5.0 | 35 | 0.0747 | 0.98 |
58
+ | 0.0008 | 6.0 | 42 | 0.0737 | 0.98 |
59
+ | 0.0007 | 7.0 | 49 | 0.0721 | 0.98 |
60
+ | 0.0006 | 8.0 | 56 | 0.0702 | 0.98 |
61
+ | 0.0006 | 9.0 | 63 | 0.0691 | 0.98 |
62
+ | 0.0006 | 10.0 | 70 | 0.0687 | 0.98 |
63
 
64
 
65
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:93d36c5262d868af38191821437bcf0be5502cff2eff519c98cca915770cab11
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d54a9a472fddd5cb8893d86650b5be4f8e48cd8139a49fba0f4649a5f7e1edf8
3
  size 267832560
tokenizer.json CHANGED
@@ -1,7 +1,19 @@
1
  {
2
  "version": "1.0",
3
- "truncation": null,
4
- "padding": null,
 
 
 
 
 
 
 
 
 
 
 
 
5
  "added_tokens": [
6
  {
7
  "id": 0,
 
1
  {
2
  "version": "1.0",
3
+ "truncation": {
4
+ "direction": "Right",
5
+ "max_length": 512,
6
+ "strategy": "LongestFirst",
7
+ "stride": 0
8
+ },
9
+ "padding": {
10
+ "strategy": "BatchLongest",
11
+ "direction": "Right",
12
+ "pad_to_multiple_of": null,
13
+ "pad_id": 0,
14
+ "pad_type_id": 0,
15
+ "pad_token": "[PAD]"
16
+ },
17
  "added_tokens": [
18
  {
19
  "id": 0,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:66f920c442f79828f98fdc2c817f96a2aa830fffc00e222e1a19d14d12c0ad3c
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e08a4b9af30d0d3b76e706671ee52deeedadda6449a5281d0fb7f83973b22185
3
  size 5368