zera09 commited on
Commit
ae98761
·
verified ·
1 Parent(s): 2cffd1e

End of training

Browse files
README.md ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: SpanBERT/spanbert-large-cased
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - f1
7
+ - accuracy
8
+ model-index:
9
+ - name: sapect_complaint_spanbert
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # sapect_complaint_spanbert
17
+
18
+ This model is a fine-tuned version of [SpanBERT/spanbert-large-cased](https://huggingface.co/SpanBERT/spanbert-large-cased) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.1905
21
+ - F1: 0.7365
22
+ - Roc Auc: 0.8250
23
+ - Accuracy: 0.5052
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 2e-05
43
+ - train_batch_size: 8
44
+ - eval_batch_size: 8
45
+ - seed: 42
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: linear
48
+ - num_epochs: 5
49
+
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
53
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|
54
+ | 0.284 | 1.0 | 582 | 0.2261 | 0.6543 | 0.7733 | 0.3720 |
55
+ | 0.2063 | 2.0 | 1164 | 0.1965 | 0.7113 | 0.8003 | 0.4338 |
56
+ | 0.1822 | 3.0 | 1746 | 0.1935 | 0.7197 | 0.8062 | 0.4416 |
57
+ | 0.1632 | 4.0 | 2328 | 0.1918 | 0.7325 | 0.8210 | 0.4897 |
58
+ | 0.1408 | 5.0 | 2910 | 0.1905 | 0.7365 | 0.8250 | 0.5052 |
59
+
60
+
61
+ ### Framework versions
62
+
63
+ - Transformers 4.41.1
64
+ - Pytorch 1.13.1+cu117
65
+ - Datasets 2.19.1
66
+ - Tokenizers 0.19.1
config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "SpanBERT/spanbert-large-cased",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "directionality": "bidi",
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 1024,
12
+ "id2label": {
13
+ "0": "Over-all_Complaint Label",
14
+ "1": "provider_response",
15
+ "2": "misc_info",
16
+ "3": "credential_error",
17
+ "4": "net_banking_issue",
18
+ "5": "consumer_safety",
19
+ "6": "review",
20
+ "7": "no_aspects",
21
+ "8": "general_query",
22
+ "9": "financial_situation",
23
+ "10": "financial_info"
24
+ },
25
+ "initializer_range": 0.02,
26
+ "intermediate_size": 4096,
27
+ "label2id": {
28
+ "Over-all_Complaint Label": 0,
29
+ "consumer_safety": 5,
30
+ "credential_error": 3,
31
+ "financial_info": 10,
32
+ "financial_situation": 9,
33
+ "general_query": 8,
34
+ "misc_info": 2,
35
+ "net_banking_issue": 4,
36
+ "no_aspects": 7,
37
+ "provider_response": 1,
38
+ "review": 6
39
+ },
40
+ "layer_norm_eps": 1e-12,
41
+ "max_position_embeddings": 512,
42
+ "model_type": "bert",
43
+ "num_attention_heads": 16,
44
+ "num_hidden_layers": 24,
45
+ "pad_token_id": 0,
46
+ "position_embedding_type": "absolute",
47
+ "problem_type": "multi_label_classification",
48
+ "torch_dtype": "float32",
49
+ "transformers_version": "4.41.1",
50
+ "type_vocab_size": 2,
51
+ "use_cache": true,
52
+ "vocab_size": 28996
53
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb23dbdadbae0a5d547d440b3debba42f2699621236738090af1f58d0acca3e8
3
+ size 1334409164
runs/Jun14_19-56-47_iit-p/events.out.tfevents.1718375211.iit-p.30207.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03fe02157f994d5180806a596b9231e2b497bfbd42216771c8d90678b3c426f9
3
+ size 5400
runs/Jun14_19-58-04_iit-p/events.out.tfevents.1718375288.iit-p.30516.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f90894541a13b4ccbf8203f63384c0118fc7927c7d34f0ef37df1c58b622f35
3
+ size 4136
runs/Jun14_19-59-46_iit-p/events.out.tfevents.1718375390.iit-p.31148.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ee6f87c900683dfff30d6f98ec3f5c0c7e666b44cb73a7a91caba973e1f9610
3
+ size 4136
runs/Jun14_20-00-44_iit-p/events.out.tfevents.1718375448.iit-p.31454.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9c96265e8c4e42b3e453fc184619dfe4cfd6a53c25c2048913cb5024b1ff3a8
3
+ size 8909
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "mask_token": "[MASK]",
49
+ "model_max_length": 1000000000000000019884624838656,
50
+ "never_split": null,
51
+ "pad_token": "[PAD]",
52
+ "sep_token": "[SEP]",
53
+ "strip_accents": null,
54
+ "tokenize_chinese_chars": true,
55
+ "tokenizer_class": "BertTokenizer",
56
+ "unk_token": "[UNK]"
57
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8982f2865b57f23bb8e669eac83d989557732de460ecac5e62e98431b6bf68d5
3
+ size 4667
vocab.txt ADDED
The diff for this file is too large to render. See raw diff