food vs nonfood classifier

README.md

@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0005
+- Loss: 0.0006
 - Accuracy: 1.0
 
 ## Model description
@@ -50,16 +50,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.3714        | 1.0   | 7    | 0.0459          | 1.0      |
-| 0.0241        | 2.0   | 14   | 0.0055          | 1.0      |
-| 0.004         | 3.0   | 21   | 0.0019          | 1.0      |
-| 0.0018        | 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.0006        | 9.0   | 63   | 0.0005          | 1.0      |
-| 0.0007        | 10.0  | 70   | 0.0005          | 1.0      |
+| 0.3264        | 1.0   | 9    | 0.0895          | 0.9709   |
+| 0.0372        | 2.0   | 18   | 0.0223          | 0.9964   |
+| 0.0225        | 3.0   | 27   | 0.0022          | 1.0      |
+| 0.0023        | 4.0   | 36   | 0.0015          | 1.0      |
+| 0.0015        | 5.0   | 45   | 0.0010          | 1.0      |
+| 0.0012        | 6.0   | 54   | 0.0008          | 1.0      |
+| 0.0009        | 7.0   | 63   | 0.0007          | 1.0      |
+| 0.0008        | 8.0   | 72   | 0.0006          | 1.0      |
+| 0.0009        | 9.0   | 81   | 0.0006          | 1.0      |
+| 0.0007        | 10.0  | 90   | 0.0006          | 1.0      |
 
 
 ### Framework versions

config.json

@@ -9,13 +9,13 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "not_food",
-    "1": "food"
+    "0": "food",
+    "1": "not_food"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "food": 1,
-    "not_food": 0
+    "food": 0,
+    "not_food": 1
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5533a04b0c7aaea9f5a3bd3786729b76f0ddcb0b969901558e0fa5e270492071
+oid sha256:267d80f37cfea177eac74cec68bd4fcbcb0ed0688739820a2967ba4a9d7de782
 size 267832560

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3b91797a282363f309ca5a4aa883bcd1abcb79f1e3552e13e9e5a67b01f12ba5
+oid sha256:7925907535a140fb8a58fa3718f780e4d3dc9110f8fd83aaf0bb5a2df8492662
 size 5240