Hugging Face's logo

joadithya
/
learn_hf_food_not_food_text_classifier-distilbert-base-uncased

Text Classification
Transformers
Safetensors
distilbert
Generated from Trainer
Model card Files Community
joadithya commited on
Commit
71b825c
·
verified ·
1 Parent(s): 933379e

Uploading first iteration of food and not food text classifier model

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
@@ -18,7 +18,7 @@ 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.0006
22
  - Accuracy: 1.0
23
 
24
  ## Model description
@@ -50,16 +50,16 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
- | 0.4121 | 1.0 | 7 | 0.0661 | 1.0 |
54
- | 0.0331 | 2.0 | 14 | 0.0925 | 0.98 |
55
- | 0.0258 | 3.0 | 21 | 0.0027 | 1.0 |
56
- | 0.0026 | 4.0 | 28 | 0.0015 | 1.0 |
57
- | 0.0017 | 5.0 | 35 | 0.0010 | 1.0 |
58
- | 0.0012 | 6.0 | 42 | 0.0008 | 1.0 |
59
- | 0.001 | 7.0 | 49 | 0.0007 | 1.0 |
60
- | 0.0009 | 8.0 | 56 | 0.0006 | 1.0 |
61
- | 0.0008 | 9.0 | 63 | 0.0006 | 1.0 |
62
- | 0.0008 | 10.0 | 70 | 0.0006 | 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.0005
22
  - Accuracy: 1.0
23
 
24
  ## Model description
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 0.5101 | 1.0 | 7 | 0.3415 | 1.0 |
54
+ | 0.2153 | 2.0 | 14 | 0.0786 | 1.0 |
55
+ | 0.0353 | 3.0 | 21 | 0.0064 | 1.0 |
56
+ | 0.004 | 4.0 | 28 | 0.0141 | 1.0 |
57
+ | 0.0168 | 5.0 | 35 | 0.0011 | 1.0 |
58
+ | 0.0011 | 6.0 | 42 | 0.0007 | 1.0 |
59
+ | 0.0008 | 7.0 | 49 | 0.0006 | 1.0 |
60
+ | 0.0008 | 8.0 | 56 | 0.0006 | 1.0 |
61
+ | 0.0007 | 9.0 | 63 | 0.0005 | 1.0 |
62
+ | 0.0007 | 10.0 | 70 | 0.0005 | 1.0 |
63
 
64
 
65
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0b1ca6fce7923570bffd09b887c237946285bbd01093c5a1fa481c3465e43149
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a50f522674f757a1b67174776cc6ca644b5363633882a6b510e75816e2c13950
3
  size 267832560
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5368531d29f5bb55ed3f7e5f95ac569cdaa1df1fd144da073efcd5b758f8731e
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a62cb315eb8e0bb0a253f16648af575826296df698fe2a7d864521e4ec7833d0
3
  size 5240