mrdbourke commited on
Commit
3d1fc90
1 Parent(s): 7c9a4a6

Uploading food not food text classifier model

Browse files
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.0009
21
  - Accuracy: 1.0
22
 
23
  ## Model description
@@ -37,7 +37,7 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - learning_rate: 2e-05
41
  - train_batch_size: 32
42
  - eval_batch_size: 32
43
  - seed: 42
@@ -49,21 +49,21 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
- | 0.0129 | 1.0 | 7 | 0.0063 | 1.0 |
53
- | 0.006 | 2.0 | 14 | 0.0034 | 1.0 |
54
- | 0.0034 | 3.0 | 21 | 0.0022 | 1.0 |
55
- | 0.0025 | 4.0 | 28 | 0.0017 | 1.0 |
56
- | 0.002 | 5.0 | 35 | 0.0014 | 1.0 |
57
- | 0.0016 | 6.0 | 42 | 0.0012 | 1.0 |
58
- | 0.0015 | 7.0 | 49 | 0.0011 | 1.0 |
59
- | 0.0013 | 8.0 | 56 | 0.0010 | 1.0 |
60
- | 0.0013 | 9.0 | 63 | 0.0010 | 1.0 |
61
- | 0.0012 | 10.0 | 70 | 0.0009 | 1.0 |
62
 
63
 
64
  ### Framework versions
65
 
66
- - Transformers 4.40.2
67
  - Pytorch 2.2.0+cu121
68
  - Datasets 2.19.1
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.0000
21
  - Accuracy: 1.0
22
 
23
  ## Model description
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 0.0001
41
  - train_batch_size: 32
42
  - eval_batch_size: 32
43
  - seed: 42
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
+ | 0.0003 | 1.0 | 7 | 0.0000 | 1.0 |
53
+ | 0.0 | 2.0 | 14 | 0.0000 | 1.0 |
54
+ | 0.0 | 3.0 | 21 | 0.0000 | 1.0 |
55
+ | 0.0 | 4.0 | 28 | 0.0000 | 1.0 |
56
+ | 0.0 | 5.0 | 35 | 0.0000 | 1.0 |
57
+ | 0.0 | 6.0 | 42 | 0.0000 | 1.0 |
58
+ | 0.0 | 7.0 | 49 | 0.0000 | 1.0 |
59
+ | 0.0 | 8.0 | 56 | 0.0000 | 1.0 |
60
+ | 0.0 | 9.0 | 63 | 0.0000 | 1.0 |
61
+ | 0.0 | 10.0 | 70 | 0.0000 | 1.0 |
62
 
63
 
64
  ### Framework versions
65
 
66
+ - Transformers 4.41.2
67
  - Pytorch 2.2.0+cu121
68
  - Datasets 2.19.1
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.40.2",
32
  "vocab_size": 30522
33
  }
 
28
  "sinusoidal_pos_embds": false,
29
  "tie_weights_": true,
30
  "torch_dtype": "float32",
31
+ "transformers_version": "4.41.2",
32
  "vocab_size": 30522
33
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:03f511d119934be564812f191d577e3fa2be35e8c1d6572cffb76500f70bda26
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b3fb1098a4715d72395229ac83a425c103ed71c891ded950216bc859045acb9
3
  size 267832560
runs/Jun14_11-21-26_DESKTOP-TMAEDD2/events.out.tfevents.1718328118.DESKTOP-TMAEDD2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9670a804e16865275352074cf1c851cdde5c6628b55114b97a73a8a1c390fa38
3
+ size 10393
tokenizer.json CHANGED
@@ -6,7 +6,14 @@
6
  "strategy": "LongestFirst",
7
  "stride": 0
8
  },
9
- "padding": null,
 
 
 
 
 
 
 
10
  "added_tokens": [
11
  {
12
  "id": 0,
 
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:41bb6fec75ab1779a7f0ee9f4e857979c14cce79a907147a52deabc484156890
3
- size 5112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e13125bda4bf1dd40e138ea67c8b1342945333e7909c0d85489720b0dbbbb8d
3
+ size 5176