Model save
Browse files
README.md
ADDED
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
library_name: transformers
|
3 |
+
license: apache-2.0
|
4 |
+
base_model: sentence-transformers/all-mpnet-base-v2
|
5 |
+
tags:
|
6 |
+
- generated_from_trainer
|
7 |
+
metrics:
|
8 |
+
- f1
|
9 |
+
- accuracy
|
10 |
+
model-index:
|
11 |
+
- name: Hub-Report-20241202125641
|
12 |
+
results: []
|
13 |
+
---
|
14 |
+
|
15 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
16 |
+
should probably proofread and complete it, then remove this comment. -->
|
17 |
+
|
18 |
+
# Hub-Report-20241202125641
|
19 |
+
|
20 |
+
This model is a fine-tuned version of [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) on the None dataset.
|
21 |
+
It achieves the following results on the evaluation set:
|
22 |
+
- Loss: 0.0629
|
23 |
+
- F1: 0.9126
|
24 |
+
- Roc Auc: 0.9528
|
25 |
+
- Accuracy: 0.9099
|
26 |
+
|
27 |
+
## Model description
|
28 |
+
|
29 |
+
More information needed
|
30 |
+
|
31 |
+
## Intended uses & limitations
|
32 |
+
|
33 |
+
More information needed
|
34 |
+
|
35 |
+
## Training and evaluation data
|
36 |
+
|
37 |
+
More information needed
|
38 |
+
|
39 |
+
## Training procedure
|
40 |
+
|
41 |
+
### Training hyperparameters
|
42 |
+
|
43 |
+
The following hyperparameters were used during training:
|
44 |
+
- learning_rate: 2e-05
|
45 |
+
- train_batch_size: 8
|
46 |
+
- eval_batch_size: 8
|
47 |
+
- seed: 42
|
48 |
+
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
49 |
+
- lr_scheduler_type: linear
|
50 |
+
- num_epochs: 13
|
51 |
+
|
52 |
+
### Training results
|
53 |
+
|
54 |
+
| Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
|
55 |
+
|:-------------:|:-----:|:-----:|:---------------:|:------:|:-------:|:--------:|
|
56 |
+
| 0.3156 | 1.0 | 936 | 0.1257 | 0.7426 | 0.8057 | 0.6152 |
|
57 |
+
| 0.0977 | 2.0 | 1872 | 0.0706 | 0.8950 | 0.9376 | 0.8831 |
|
58 |
+
| 0.0526 | 3.0 | 2808 | 0.0596 | 0.9000 | 0.9442 | 0.8946 |
|
59 |
+
| 0.032 | 4.0 | 3744 | 0.0551 | 0.9081 | 0.9497 | 0.9036 |
|
60 |
+
| 0.0226 | 5.0 | 4680 | 0.0632 | 0.8951 | 0.9428 | 0.8909 |
|
61 |
+
| 0.0193 | 6.0 | 5616 | 0.0579 | 0.9098 | 0.9510 | 0.9068 |
|
62 |
+
| 0.0156 | 7.0 | 6552 | 0.0607 | 0.9086 | 0.9504 | 0.9046 |
|
63 |
+
| 0.0129 | 8.0 | 7488 | 0.0611 | 0.9118 | 0.9523 | 0.9080 |
|
64 |
+
| 0.0126 | 9.0 | 8424 | 0.0633 | 0.9114 | 0.9529 | 0.9077 |
|
65 |
+
| 0.0107 | 10.0 | 9360 | 0.0629 | 0.9126 | 0.9528 | 0.9099 |
|
66 |
+
| 0.0084 | 11.0 | 10296 | 0.0654 | 0.9091 | 0.9510 | 0.9058 |
|
67 |
+
| 0.0079 | 12.0 | 11232 | 0.0647 | 0.9100 | 0.9521 | 0.9055 |
|
68 |
+
| 0.0065 | 13.0 | 12168 | 0.0652 | 0.9102 | 0.9523 | 0.9071 |
|
69 |
+
|
70 |
+
|
71 |
+
### Framework versions
|
72 |
+
|
73 |
+
- Transformers 4.46.2
|
74 |
+
- Pytorch 2.5.1+cu121
|
75 |
+
- Datasets 3.1.0
|
76 |
+
- Tokenizers 0.20.3
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 438005960
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c3f9b4fb6a4c961f080e18be61b7368aff1bfc635281a1861c5be9530ae96e77
|
3 |
size 438005960
|
runs/Dec02_12-57-11_dd8fdf6124ff/events.out.tfevents.1733149616.dd8fdf6124ff.2341.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5629011ae91bee0daeb9f2a114e6bf028ff611c4cdc5fe29ccd0e04dffeb57c3
|
3 |
+
size 508
|
tokenizer.json
CHANGED
@@ -1,21 +1,7 @@
|
|
1 |
{
|
2 |
"version": "1.0",
|
3 |
-
"truncation":
|
4 |
-
|
5 |
-
"max_length": 256,
|
6 |
-
"strategy": "LongestFirst",
|
7 |
-
"stride": 0
|
8 |
-
},
|
9 |
-
"padding": {
|
10 |
-
"strategy": {
|
11 |
-
"Fixed": 256
|
12 |
-
},
|
13 |
-
"direction": "Right",
|
14 |
-
"pad_to_multiple_of": null,
|
15 |
-
"pad_id": 1,
|
16 |
-
"pad_type_id": 0,
|
17 |
-
"pad_token": "<pad>"
|
18 |
-
},
|
19 |
"added_tokens": [
|
20 |
{
|
21 |
"id": 0,
|
|
|
1 |
{
|
2 |
"version": "1.0",
|
3 |
+
"truncation": null,
|
4 |
+
"padding": null,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
"added_tokens": [
|
6 |
{
|
7 |
"id": 0,
|