lobrien001 commited on
Commit
db8065a
·
verified ·
1 Parent(s): 446f0d1

Model save

Browse files
Files changed (5) hide show
  1. README.md +11 -10
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. tokenizer.json +2 -16
  5. training_args.bin +2 -2
README.md CHANGED
@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.2161
24
- - Precision: 0.9547
25
- - Recall: 0.9504
26
- - F1: 0.9525
27
- - Accuracy: 0.9454
28
 
29
  ## Model description
30
 
@@ -44,8 +44,8 @@ More information needed
44
 
45
  The following hyperparameters were used during training:
46
  - learning_rate: 2e-05
47
- - train_batch_size: 8
48
- - eval_batch_size: 8
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
@@ -55,12 +55,13 @@ The following hyperparameters were used during training:
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
- | 0.084 | 7.46 | 1000 | 0.2167 | 0.9544 | 0.9523 | 0.9534 | 0.9460 |
 
59
 
60
 
61
  ### Framework versions
62
 
63
- - Transformers 4.39.3
64
- - Pytorch 2.1.2
65
  - Datasets 2.18.0
66
  - Tokenizers 0.15.2
 
20
 
21
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.2658
24
+ - Precision: 0.9566
25
+ - Recall: 0.9493
26
+ - F1: 0.9529
27
+ - Accuracy: 0.9465
28
 
29
  ## Model description
30
 
 
44
 
45
  The following hyperparameters were used during training:
46
  - learning_rate: 2e-05
47
+ - train_batch_size: 4
48
+ - eval_batch_size: 4
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
 
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
+ | 0.1398 | 3.73 | 1000 | 0.2144 | 0.9545 | 0.9406 | 0.9475 | 0.9405 |
59
+ | 0.0541 | 7.46 | 2000 | 0.2612 | 0.9543 | 0.9494 | 0.9518 | 0.9454 |
60
 
61
 
62
  ### Framework versions
63
 
64
+ - Transformers 4.36.2
65
+ - Pytorch 2.2.1+cu121
66
  - Datasets 2.18.0
67
  - Tokenizers 0.15.2
config.json CHANGED
@@ -32,7 +32,7 @@
32
  "pad_token_id": 1,
33
  "position_embedding_type": "absolute",
34
  "torch_dtype": "float32",
35
- "transformers_version": "4.39.3",
36
  "type_vocab_size": 1,
37
  "use_cache": true,
38
  "vocab_size": 50265
 
32
  "pad_token_id": 1,
33
  "position_embedding_type": "absolute",
34
  "torch_dtype": "float32",
35
+ "transformers_version": "4.36.2",
36
  "type_vocab_size": 1,
37
  "use_cache": true,
38
  "vocab_size": 50265
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b11fe42280ef81c7e327fc3a938cfb4054b8630eec087a94108914ad842f00f8
3
  size 496256392
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23ddf290a1a127f87471cc7685bd809e088128e4be8ad94c0eeccc9cc0fea24d
3
  size 496256392
tokenizer.json CHANGED
@@ -1,21 +1,7 @@
1
  {
2
  "version": "1.0",
3
- "truncation": {
4
- "direction": "Right",
5
- "max_length": 512,
6
- "strategy": "LongestFirst",
7
- "stride": 0
8
- },
9
- "padding": {
10
- "strategy": {
11
- "Fixed": 512
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,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f37c58efd725d2f3e50c51eccd327dff51ef1c1e694455990c13c4d07f347330
3
- size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:480c7029a97be64fec2f18b786c198cc44c8bc6d0b0ccc9282d1c8217963beda
3
+ size 4728