SudiptoPramanik/Mail_RewardModel_RobertaBase

README.md

@@ -19,7 +19,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0066
+- Loss: 0.0064
 - F1: 1.0
 - Roc Auc: 1.0
 - Accuracy: 1.0
@@ -53,11 +53,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | F1  | Roc Auc | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---:|:-------:|:--------:|
-| No log        | 1.0   | 63   | 0.0066          | 1.0 | 1.0     | 1.0      |
-| 0.1435        | 2.0   | 126  | 0.0028          | 1.0 | 1.0     | 1.0      |
-| 0.1435        | 3.0   | 189  | 0.0019          | 1.0 | 1.0     | 1.0      |
+| No log        | 1.0   | 63   | 0.0064          | 1.0 | 1.0     | 1.0      |
+| 0.1495        | 2.0   | 126  | 0.0028          | 1.0 | 1.0     | 1.0      |
+| 0.1495        | 3.0   | 189  | 0.0020          | 1.0 | 1.0     | 1.0      |
 | 0.0035        | 4.0   | 252  | 0.0016          | 1.0 | 1.0     | 1.0      |
-| 0.0023        | 5.0   | 315  | 0.0015          | 1.0 | 1.0     | 1.0      |
+| 0.0024        | 5.0   | 315  | 0.0015          | 1.0 | 1.0     | 1.0      |
 
 
 ### Framework versions

config.json

@@ -11,14 +11,14 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "Bad",
-    "1": "Good"
+    "0": "spam",
+    "1": "not spam"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Bad": 0,
-    "Good": 1
+    "not spam": 1,
+    "spam": 0
   },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:01f69b702a2804a07ba318ea5285e6f60823412bbf99c476c5bea53d67f8eee2
+oid sha256:fc34758264abaf139eef1509088ae68e02a6955548e98e9828b8471f41101dab
 size 498612824

runs/Jan05_15-02-25_3f7c07b78d44/events.out.tfevents.1736089351.3f7c07b78d44.1813.0

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

runs/Jan05_15-02-25_3f7c07b78d44/events.out.tfevents.1736089930.3f7c07b78d44.1813.1

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e6bef81fe1adfcd7f019233a829f63de1dabdb86c5d11dee6750db616c9504ff
+oid sha256:8eb3b3d4ebc6dfb7b336cd28c596b88dfe2556a1f9cb82177494becdf3c31344
 size 5368