Sbudo text classifier

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/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.1424
-- Accuracy: 0.9698
+- Loss: 0.0823
+- Accuracy: 0.9903
 
 ## Model description
 
@@ -50,16 +50,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.4879        | 1.0   | 25   | 0.2264          | 0.9196   |
-| 0.1452        | 2.0   | 50   | 0.2094          | 0.9447   |
-| 0.0837        | 3.0   | 75   | 0.0876          | 0.9749   |
-| 0.0281        | 4.0   | 100  | 0.1444          | 0.9648   |
-| 0.0127        | 5.0   | 125  | 0.1225          | 0.9698   |
-| 0.0008        | 6.0   | 150  | 0.1622          | 0.9698   |
-| 0.0005        | 7.0   | 175  | 0.1584          | 0.9698   |
-| 0.0004        | 8.0   | 200  | 0.1448          | 0.9698   |
-| 0.0003        | 9.0   | 225  | 0.1428          | 0.9698   |
-| 0.0003        | 10.0  | 250  | 0.1424          | 0.9698   |
+| 0.5879        | 1.0   | 26   | 0.1729          | 0.9420   |
+| 0.1047        | 2.0   | 52   | 0.1249          | 0.9758   |
+| 0.042         | 3.0   | 78   | 0.0938          | 0.9807   |
+| 0.0022        | 4.0   | 104  | 0.0756          | 0.9903   |
+| 0.0086        | 5.0   | 130  | 0.0767          | 0.9903   |
+| 0.0005        | 6.0   | 156  | 0.0789          | 0.9903   |
+| 0.0004        | 7.0   | 182  | 0.0804          | 0.9903   |
+| 0.0003        | 8.0   | 208  | 0.0815          | 0.9903   |
+| 0.0003        | 9.0   | 234  | 0.0821          | 0.9903   |
+| 0.0003        | 10.0  | 260  | 0.0823          | 0.9903   |
 
 
 ### Framework versions

config.json

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

model.safetensors

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:21a5115fdf89790f78c73aee5603148cfc3fabee949f2fdbbe066d8ea82e5df4
-size 5368
+oid sha256:60582601f476aff6eb4fc3a1532fae6474f034e3c645d1ba7329eed947564fdd
+size 5304