redscroll commited on
Commit
237aba4
·
1 Parent(s): b36fd18

redscroll/roberta-finetuned-domains

Browse files
Files changed (4) hide show
  1. README.md +9 -9
  2. config.json +13 -13
  3. pytorch_model.bin +1 -1
  4. training_args.bin +1 -1
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 [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.9364
22
- - F1: 0.3923
23
- - Roc Auc: 0.6283
24
- - Accuracy: 0.3298
25
 
26
  ## Model description
27
 
@@ -53,15 +53,15 @@ The following hyperparameters were used during training:
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
55
  |:-------------:|:-----:|:-----:|:---------------:|:------:|:-------:|:--------:|
56
- | 0.1009 | 1.0 | 4639 | 0.6919 | 0.3453 | 0.6010 | 0.2721 |
57
- | 0.0853 | 2.0 | 9278 | 0.7979 | 0.3516 | 0.6044 | 0.2856 |
58
- | 0.0562 | 3.0 | 13917 | 0.9364 | 0.3923 | 0.6283 | 0.3298 |
59
- | 0.0109 | 4.0 | 18556 | 1.0486 | 0.3707 | 0.6156 | 0.3036 |
60
 
61
 
62
  ### Framework versions
63
 
64
- - Transformers 4.33.2
65
  - Pytorch 2.0.1+cu117
66
  - Datasets 2.14.5
67
  - Tokenizers 0.13.3
 
18
 
19
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.8502
22
+ - F1: 0.3317
23
+ - Roc Auc: 0.5777
24
+ - Accuracy: 0.1883
25
 
26
  ## Model description
27
 
 
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
55
  |:-------------:|:-----:|:-----:|:---------------:|:------:|:-------:|:--------:|
56
+ | 0.2412 | 1.0 | 24797 | 0.8502 | 0.3317 | 0.5777 | 0.1883 |
57
+ | 0.129 | 2.0 | 49594 | 0.9576 | 0.3219 | 0.5724 | 0.1962 |
58
+ | 0.1072 | 3.0 | 74391 | 1.2442 | 0.3260 | 0.5718 | 0.1906 |
59
+ | 0.0422 | 4.0 | 99188 | 1.4241 | 0.3259 | 0.5723 | 0.1927 |
60
 
61
 
62
  ### Framework versions
63
 
64
+ - Transformers 4.33.1
65
  - Pytorch 2.0.1+cu117
66
  - Datasets 2.14.5
67
  - Tokenizers 0.13.3
config.json CHANGED
@@ -10,22 +10,22 @@
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
- "0": "Access to Resources",
14
- "1": "Knowledge & Beliefs",
15
- "2": "Law & Policy",
16
- "3": "Human Dignity",
17
- "4": "Roles & Responsibilities",
18
- "5": "Power & Decision-Making"
19
  },
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "Access to Resources": 0,
24
- "Human Dignity": 3,
25
- "Knowledge & Beliefs": 1,
26
- "Law & Policy": 2,
27
- "Power & Decision-Making": 5,
28
- "Roles & Responsibilities": 4
29
  },
30
  "layer_norm_eps": 1e-12,
31
  "max_position_embeddings": 512,
@@ -36,7 +36,7 @@
36
  "position_embedding_type": "absolute",
37
  "problem_type": "multi_label_classification",
38
  "torch_dtype": "float32",
39
- "transformers_version": "4.33.2",
40
  "type_vocab_size": 2,
41
  "use_cache": true,
42
  "vocab_size": 30522
 
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
+ "0": "Power & Decision-Making",
14
+ "1": "Access to Resources",
15
+ "2": "Human Dignity",
16
+ "3": "Knowledge & Beliefs",
17
+ "4": "Law & Policy",
18
+ "5": "Roles & Responsibilities"
19
  },
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
+ "Access to Resources": 1,
24
+ "Human Dignity": 2,
25
+ "Knowledge & Beliefs": 3,
26
+ "Law & Policy": 4,
27
+ "Power & Decision-Making": 0,
28
+ "Roles & Responsibilities": 5
29
  },
30
  "layer_norm_eps": 1e-12,
31
  "max_position_embeddings": 512,
 
36
  "position_embedding_type": "absolute",
37
  "problem_type": "multi_label_classification",
38
  "torch_dtype": "float32",
39
+ "transformers_version": "4.33.1",
40
  "type_vocab_size": 2,
41
  "use_cache": true,
42
  "vocab_size": 30522
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c87ac7be13f1f1408a0e3d59f5574889da93e5aeb85957b936a2d9a4f5a9e41
3
  size 438015793
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a314b90b6e9a237652f6fb819d2528df6caec367adbe200633370de6e057249
3
  size 438015793
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0ba40ddf51bd14597942e02a97a2b795945fcd3cecec04ee6c39a4472b03fcc0
3
  size 4091
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62c9967b28014ed511609a62b04668bf91dfb9839e81884c2ce7b6fb7a2910e8
3
  size 4091