Hugging Face's logo

mrdbourke
/
learn_hf_food_not_food_text_classifier-distilbert-base-uncased

Text Classification
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
mrdbourke commited on
Commit
1d40cf6
•
1 Parent(s): 0307827

Uploading food not food text classifier model (putting it all together!)

Browse files
Files changed (3) hide show
  1. README.md +11 -11
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
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.0006
21
  - Accuracy: 1.0
22
 
23
  ## Model description
@@ -49,16 +49,16 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
- | 0.4116 | 1.0 | 7 | 0.0689 | 1.0 |
53
- | 0.032 | 2.0 | 14 | 0.0075 | 1.0 |
54
- | 0.0055 | 3.0 | 21 | 0.0026 | 1.0 |
55
- | 0.0022 | 4.0 | 28 | 0.0015 | 1.0 |
56
- | 0.0013 | 5.0 | 35 | 0.0010 | 1.0 |
57
- | 0.0011 | 6.0 | 42 | 0.0008 | 1.0 |
58
- | 0.0009 | 7.0 | 49 | 0.0007 | 1.0 |
59
- | 0.0007 | 8.0 | 56 | 0.0007 | 1.0 |
60
- | 0.0007 | 9.0 | 63 | 0.0006 | 1.0 |
61
- | 0.0007 | 10.0 | 70 | 0.0006 | 1.0 |
62
 
63
 
64
  ### Framework versions
 
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.3943 | 1.0 | 7 | 0.0926 | 1.0 |
53
+ | 0.0403 | 2.0 | 14 | 0.0067 | 1.0 |
54
+ | 0.0047 | 3.0 | 21 | 0.0021 | 1.0 |
55
+ | 0.0018 | 4.0 | 28 | 0.0011 | 1.0 |
56
+ | 0.001 | 5.0 | 35 | 0.0008 | 1.0 |
57
+ | 0.0008 | 6.0 | 42 | 0.0006 | 1.0 |
58
+ | 0.0007 | 7.0 | 49 | 0.0006 | 1.0 |
59
+ | 0.0006 | 8.0 | 56 | 0.0005 | 1.0 |
60
+ | 0.0005 | 9.0 | 63 | 0.0005 | 1.0 |
61
+ | 0.0006 | 10.0 | 70 | 0.0005 | 1.0 |
62
 
63
 
64
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d219389cd95bc8d404de998a3ea6578e94c640c371fa28191125fa297cbc28a9
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80c8534de8b9f568be169c5830e7a70d3005557c6c02a3dc65504ec8c8e7a139
3
  size 267832560
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1d925bebc9790cc44e681a124688cb75918341a8fbec451c8c5a1f1b1427bcc9
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5dfe04edd9576a9f88355e365cc0bcc155b2f18b61030a2e230fb828562a826c
3
  size 5176