RajuEEE/RewardModel_RobertaBase
Browse files
README.md
CHANGED
@@ -18,10 +18,10 @@ should probably proofread and complete it, then remove this comment. -->
|
|
18 |
|
19 |
This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
|
20 |
It achieves the following results on the evaluation set:
|
21 |
-
- Loss: 0.
|
22 |
-
- F1: 0.
|
23 |
-
- Roc Auc: 0.
|
24 |
-
- Accuracy: 0.
|
25 |
|
26 |
## Model description
|
27 |
|
@@ -52,14 +52,14 @@ The following hyperparameters were used during training:
|
|
52 |
|
53 |
| Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
|
54 |
|:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|
|
55 |
-
| No log | 1.0 |
|
56 |
-
|
|
57 |
-
|
|
58 |
|
59 |
|
60 |
### Framework versions
|
61 |
|
62 |
-
- Transformers 4.
|
63 |
-
- Pytorch 2.
|
64 |
-
- Datasets 2.
|
65 |
-
- Tokenizers 0.
|
|
|
18 |
|
19 |
This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
|
20 |
It achieves the following results on the evaluation set:
|
21 |
+
- Loss: 0.6783
|
22 |
+
- F1: 0.7879
|
23 |
+
- Roc Auc: 0.825
|
24 |
+
- Accuracy: 0.65
|
25 |
|
26 |
## Model description
|
27 |
|
|
|
52 |
|
53 |
| Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
|
54 |
|:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|
|
55 |
+
| No log | 1.0 | 10 | 0.6885 | 0.2222 | 0.5625 | 0.125 |
|
56 |
+
| No log | 2.0 | 20 | 0.6844 | 0.6207 | 0.725 | 0.45 |
|
57 |
+
| No log | 3.0 | 30 | 0.6783 | 0.7879 | 0.825 | 0.65 |
|
58 |
|
59 |
|
60 |
### Framework versions
|
61 |
|
62 |
+
- Transformers 4.41.2
|
63 |
+
- Pytorch 2.3.0+cu121
|
64 |
+
- Datasets 2.20.0
|
65 |
+
- Tokenizers 0.19.1
|
config.json
CHANGED
@@ -29,7 +29,7 @@
|
|
29 |
"position_embedding_type": "absolute",
|
30 |
"problem_type": "multi_label_classification",
|
31 |
"torch_dtype": "float32",
|
32 |
-
"transformers_version": "4.
|
33 |
"type_vocab_size": 1,
|
34 |
"use_cache": true,
|
35 |
"vocab_size": 50265
|
|
|
29 |
"position_embedding_type": "absolute",
|
30 |
"problem_type": "multi_label_classification",
|
31 |
"torch_dtype": "float32",
|
32 |
+
"transformers_version": "4.41.2",
|
33 |
"type_vocab_size": 1,
|
34 |
"use_cache": true,
|
35 |
"vocab_size": 50265
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 498612824
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3798861170f9d71666f8d027e4a6f165c954832666ab1753fa9a011fa88be543
|
3 |
size 498612824
|
runs/Jun15_04-45-17_cebb3ca22ba1/events.out.tfevents.1718426719.cebb3ca22ba1.153.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:131b8ae102cb5b54c337b612295d391a505bb4954ce88d56555f82007426ae72
|
3 |
+
size 6564
|
runs/Jun15_04-45-17_cebb3ca22ba1/events.out.tfevents.1718426833.cebb3ca22ba1.153.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eda870c223de85e72b993b626fc10ceceb906fa2966ac3b4dc1313d5c6bb931c
|
3 |
+
size 500
|
tokenizer.json
CHANGED
@@ -83,6 +83,7 @@
|
|
83 |
"end_of_word_suffix": "",
|
84 |
"fuse_unk": false,
|
85 |
"byte_fallback": false,
|
|
|
86 |
"vocab": {
|
87 |
"<s>": 0,
|
88 |
"<pad>": 1,
|
|
|
83 |
"end_of_word_suffix": "",
|
84 |
"fuse_unk": false,
|
85 |
"byte_fallback": false,
|
86 |
+
"ignore_merges": false,
|
87 |
"vocab": {
|
88 |
"<s>": 0,
|
89 |
"<pad>": 1,
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0f703e3b72ae27ef4f5e9a269353350f65f2c64b4d9943155ecf0cc2a7cb531d
|
3 |
+
size 5112
|