KingTechnician/distilbert-base-uncased-feverous-nei-binary-v2

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: 2.1618
-- Accuracy: 0.7621
+- Loss: 0.6157
+- Accuracy: 0.6103
 
 ## Model description
 
@@ -44,18 +44,15 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 6
+- num_epochs: 3
 
 ### Training results
 
-| Training Loss | Epoch | Step  | Validation Loss | Accuracy |
-|:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 0.8395        | 1.0   | 3119  | 1.0347          | 0.7669   |
-| 0.7849        | 2.0   | 6238  | 1.4335          | 0.7678   |
-| 0.6369        | 3.0   | 9357  | 1.1698          | 0.7620   |
-| 0.4997        | 4.0   | 12476 | 1.5501          | 0.7607   |
-| 0.3769        | 5.0   | 15595 | 1.7184          | 0.7489   |
-| 0.2832        | 6.0   | 18714 | 2.1618          | 0.7621   |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| No log        | 1.0   | 199  | 0.6703          | 0.5198   |
+| No log        | 2.0   | 398  | 0.5498          | 0.6734   |
+| 0.6403        | 3.0   | 597  | 0.6157          | 0.6103   |
 
 
 ### Framework versions

config.json

@@ -9,21 +9,20 @@
   "dtype": "float32",
   "hidden_dim": 3072,
   "id2label": {
-    "0": "SUPPORTS",
-    "1": "REFUTES",
-    "2": "NOT ENOUGH INFO"
+    "0": "ENOUGH INFO",
+    "1": "NOT ENOUGH INFO"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "NOT ENOUGH INFO": 2,
-    "REFUTES": 1,
-    "SUPPORTS": 0
+    "ENOUGH INFO": 0,
+    "NOT ENOUGH INFO": 1
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
+  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:01248e0b9f9271d408c6e860b7c92b0a9a87dbcb20428ce919170f544f43cf96
-size 267835644
+oid sha256:41f71184e0ea2e5e7dbefc48accfcba87eecd3555331c30788310399ea852444
+size 267832560

runs/Dec01_22-38-04_1385cd96f5cf/events.out.tfevents.1764628686.1385cd96f5cf.2474.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23028a03e6ada1a85549cbd04192c782df8d4eb7bc5a01875e59c7c66908e7fd
+size 6574

training_args.bin

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