Uploading food not food text classifier model

README.md

@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3873
-- Accuracy: 0.885
+- Loss: 0.3765
+- Accuracy: 0.895
 
 ## Model description
 
@@ -50,16 +50,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.2909        | 1.0   | 32   | 1.1368          | 0.745    |
-| 0.8996        | 2.0   | 64   | 0.7569          | 0.83     |
-| 0.5752        | 3.0   | 96   | 0.5593          | 0.855    |
-| 0.4121        | 4.0   | 128  | 0.4700          | 0.885    |
-| 0.3201        | 5.0   | 160  | 0.4398          | 0.875    |
-| 0.2773        | 6.0   | 192  | 0.4191          | 0.885    |
-| 0.2301        | 7.0   | 224  | 0.4023          | 0.885    |
-| 0.2093        | 8.0   | 256  | 0.3955          | 0.885    |
-| 0.1937        | 9.0   | 288  | 0.3886          | 0.885    |
-| 0.1912        | 10.0  | 320  | 0.3873          | 0.885    |
+| 1.2846        | 1.0   | 32   | 1.1305          | 0.745    |
+| 0.9125        | 2.0   | 64   | 0.7700          | 0.845    |
+| 0.5789        | 3.0   | 96   | 0.5487          | 0.86     |
+| 0.4028        | 4.0   | 128  | 0.4588          | 0.87     |
+| 0.3067        | 5.0   | 160  | 0.4303          | 0.87     |
+| 0.2694        | 6.0   | 192  | 0.4051          | 0.87     |
+| 0.2243        | 7.0   | 224  | 0.3891          | 0.89     |
+| 0.2052        | 8.0   | 256  | 0.3834          | 0.89     |
+| 0.1907        | 9.0   | 288  | 0.3778          | 0.89     |
+| 0.1891        | 10.0  | 320  | 0.3765          | 0.895    |
 
 
 ### Framework versions

config.json

@@ -9,10 +9,10 @@
   "dtype": "float32",
   "hidden_dim": 3072,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3"
+    "0": "World",
+    "1": "Sports",
+    "2": "Business",
+    "3": "Sci/Tech"
   },
   "initializer_range": 0.02,
   "label2id": {

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1d5ec65bc9a43b6d23720c625c6d017b4cb10360554791535776c5a5efba7dcd
+oid sha256:9af7d522470682b4a2af1cf8d9bbd3968b9556ce1d19f0d3a6dc7509c1751f56
 size 267838720

training_args.bin

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