mrdbourke commited on
Commit
2d25a2f
1 Parent(s): ecda646

Uploading food not food text classifier model

Browse files
Files changed (5) hide show
  1. README.md +13 -13
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. tokenizer.json +14 -2
  5. training_args.bin +2 -2
README.md CHANGED
@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.0007
21
  - Accuracy: 1.0
22
 
23
  ## Model description
@@ -49,21 +49,21 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
- | 0.3725 | 1.0 | 7 | 0.0679 | 1.0 |
53
- | 0.0283 | 2.0 | 14 | 0.0092 | 1.0 |
54
- | 0.0047 | 3.0 | 21 | 0.0049 | 1.0 |
55
- | 0.002 | 4.0 | 28 | 0.0021 | 1.0 |
56
- | 0.0012 | 5.0 | 35 | 0.0013 | 1.0 |
57
- | 0.0009 | 6.0 | 42 | 0.0010 | 1.0 |
58
- | 0.0008 | 7.0 | 49 | 0.0008 | 1.0 |
59
- | 0.0007 | 8.0 | 56 | 0.0008 | 1.0 |
60
- | 0.0007 | 9.0 | 63 | 0.0007 | 1.0 |
61
- | 0.0007 | 10.0 | 70 | 0.0007 | 1.0 |
62
 
63
 
64
  ### Framework versions
65
 
66
- - Transformers 4.43.2
67
- - Pytorch 2.4.0+cu121
68
  - Datasets 2.20.0
69
  - Tokenizers 0.19.1
 
17
 
18
  This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.0005
21
  - Accuracy: 1.0
22
 
23
  ## Model description
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
+ | 0.4242 | 1.0 | 7 | 0.1441 | 1.0 |
53
+ | 0.0955 | 2.0 | 14 | 0.0094 | 1.0 |
54
+ | 0.0074 | 3.0 | 21 | 0.0024 | 1.0 |
55
+ | 0.0022 | 4.0 | 28 | 0.0012 | 1.0 |
56
+ | 0.0012 | 5.0 | 35 | 0.0008 | 1.0 |
57
+ | 0.0008 | 6.0 | 42 | 0.0006 | 1.0 |
58
+ | 0.0007 | 7.0 | 49 | 0.0005 | 1.0 |
59
+ | 0.0006 | 8.0 | 56 | 0.0005 | 1.0 |
60
+ | 0.0006 | 9.0 | 63 | 0.0005 | 1.0 |
61
+ | 0.0006 | 10.0 | 70 | 0.0005 | 1.0 |
62
 
63
 
64
  ### Framework versions
65
 
66
+ - Transformers 4.42.4
67
+ - Pytorch 2.3.1+cu121
68
  - Datasets 2.20.0
69
  - Tokenizers 0.19.1
config.json CHANGED
@@ -28,6 +28,6 @@
28
  "sinusoidal_pos_embds": false,
29
  "tie_weights_": true,
30
  "torch_dtype": "float32",
31
- "transformers_version": "4.43.2",
32
  "vocab_size": 30522
33
  }
 
28
  "sinusoidal_pos_embds": false,
29
  "tie_weights_": true,
30
  "torch_dtype": "float32",
31
+ "transformers_version": "4.42.4",
32
  "vocab_size": 30522
33
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9520b1be2ae7d19b5b2eaaf12526af26533869d3f8c250183ea7dbf74f713afc
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91260d38856871115f09a7beecc7af73567d2b49a59c648dd979b7aa234ea1c3
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:c0f7c3660fbea763117129af2c6bb31b9270efe866d7ce096af4ff834923877b
3
- size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:859e62d73d803560d74159bbd43f66204d63a3f1b5c5cc2f12eef2fd79362fc3
3
+ size 5176