mrdbourke
/

learn_hf_food_not_food_text_classifier-distilbert-base-uncased

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

mrdbourke commited on Jul 26, 2024

Commit

8a8a8af

·

verified ·

1 Parent(s): 5ddc1cf

Uploading food not food text classifier model

Files changed (4) hide show

README.md +11 -11
config.json +1 -1
model.safetensors +1 -1
training_args.bin +2 -2

README.md CHANGED Viewed

@@ -49,21 +49,21 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.4182        | 1.0   | 7    | 0.0619          | 1.0      |
-| 0.0278        | 2.0   | 14   | 0.0064          | 1.0      |
-| 0.0046        | 3.0   | 21   | 0.0020          | 1.0      |
-| 0.0019        | 4.0   | 28   | 0.0011          | 1.0      |
-| 0.0012        | 5.0   | 35   | 0.0008          | 1.0      |
-| 0.0009        | 6.0   | 42   | 0.0006          | 1.0      |
 | 0.0008        | 7.0   | 49   | 0.0006          | 1.0      |
-| 0.0007        | 8.0   | 56   | 0.0005          | 1.0      |
-| 0.0007        | 9.0   | 63   | 0.0005          | 1.0      |
-| 0.0007        | 10.0  | 70   | 0.0005          | 1.0      |
 ### Framework versions
-- Transformers 4.42.4
-- Pytorch 2.3.1+cu121
 - Datasets 2.20.0
 - Tokenizers 0.19.1

 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.3281        | 1.0   | 7    | 0.0396          | 1.0      |
+| 0.0192        | 2.0   | 14   | 0.0056          | 1.0      |
+| 0.0037        | 3.0   | 21   | 0.0020          | 1.0      |
+| 0.0017        | 4.0   | 28   | 0.0012          | 1.0      |
+| 0.0011        | 5.0   | 35   | 0.0009          | 1.0      |
+| 0.0008        | 6.0   | 42   | 0.0007          | 1.0      |
 | 0.0008        | 7.0   | 49   | 0.0006          | 1.0      |
+| 0.0007        | 8.0   | 56   | 0.0006          | 1.0      |
+| 0.0006        | 9.0   | 63   | 0.0005          | 1.0      |
+| 0.0006        | 10.0  | 70   | 0.0005          | 1.0      |
 ### Framework versions
+- Transformers 4.43.2
+- Pytorch 2.4.0+cu121
 - Datasets 2.20.0
 - Tokenizers 0.19.1

config.json CHANGED Viewed

@@ -28,6 +28,6 @@
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.42.4",
   "vocab_size": 30522
 }

   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
   "torch_dtype": "float32",
+  "transformers_version": "4.43.2",
   "vocab_size": 30522
 }

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e20400858fcb16c7f45c32943076055457080b26e0969965b0662aae3eaa0f9
 size 267832560

 version https://git-lfs.github.com/spec/v1
+oid sha256:76391ae9c0ded0e389c5d884751067252c852c50543405c0f9d734f16d781a20
 size 267832560

training_args.bin CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:da3c101c6f3791373bf1b26fd9d22a9a3929d41277488946c5b7a2627b2faf43
-size 5176

 version https://git-lfs.github.com/spec/v1
+oid sha256:674e214664490094f43cf39d4629ce496eb5caa0c45d1c311dd3cbbbc81ada4f
+size 5240