Hugging Face's logo

cmcmaster
/
gca_tab_ckpts

Text Classification
Transformers
Safetensors
deberta-v2
Generated from Trainer
Model card Files
xet
Community
cmcmaster commited on
Commit
21af4b5
·
1 Parent(s): 44483d4

austin/gca_tab

Browse files
Files changed (9) hide show
  1. README.md +82 -0
  2. added_tokens.json +3 -0
  3. config.json +48 -0
  4. model.safetensors +3 -0
  5. special_tokens_map.json +9 -0
  6. spm.model +3 -0
  7. tokenizer.json +0 -0
  8. tokenizer_config.json +58 -0
  9. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: austin/mimic-pubmed-deberta-small
3
+ tags:
4
+ - generated_from_trainer
5
+ model-index:
6
+ - name: gca_tab_ckpts
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
12
+
13
+ # gca_tab_ckpts
14
+
15
+ This model is a fine-tuned version of [austin/mimic-pubmed-deberta-small](https://huggingface.co/austin/mimic-pubmed-deberta-small) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+ - Loss: 0.0831
18
+ - F1 Adventitial inflammation: 0.4
19
+ - F1 Gca: 0.8889
20
+ - F1 Giant cells: 1.0
21
+ - F1 Intimal hyperplasia: 0.8
22
+ - Acc Adventitial inflammation: 0.9211
23
+ - Acc Gca: 0.9737
24
+ - Acc Giant cells: 1.0
25
+ - Acc Intimal hyperplasia: 0.9737
26
+ - Auc Adventitial inflammation: 0.9306
27
+ - Auc Gca: 0.9939
28
+ - Auc Giant cells: 1.0
29
+ - Auc Intimal hyperplasia: 0.9861
30
+
31
+ ## Model description
32
+
33
+ More information needed
34
+
35
+ ## Intended uses & limitations
36
+
37
+ More information needed
38
+
39
+ ## Training and evaluation data
40
+
41
+ More information needed
42
+
43
+ ## Training procedure
44
+
45
+ ### Training hyperparameters
46
+
47
+ The following hyperparameters were used during training:
48
+ - learning_rate: 0.0001
49
+ - train_batch_size: 32
50
+ - eval_batch_size: 32
51
+ - seed: 42
52
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
+ - lr_scheduler_type: linear
54
+ - num_epochs: 15
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss | F1 Adventitial inflammation | F1 Gca | F1 Giant cells | F1 Intimal hyperplasia | Acc Adventitial inflammation | Acc Gca | Acc Giant cells | Acc Intimal hyperplasia | Auc Adventitial inflammation | Auc Gca | Auc Giant cells | Auc Intimal hyperplasia |
59
+ |:-------------:|:-----:|:----:|:---------------:|:---------------------------:|:------:|:--------------:|:----------------------:|:----------------------------:|:-------:|:---------------:|:-----------------------:|:----------------------------:|:-------:|:---------------:|:-----------------------:|
60
+ | 0.5304 | 1.0 | 11 | 0.2758 | 0.0 | 0.0 | 0.0 | 0.0 | 0.9474 | 0.8684 | 0.9211 | 0.9474 | 0.0417 | 0.4182 | 0.0952 | 0.9444 |
61
+ | 0.4395 | 2.0 | 22 | 0.2832 | 0.0 | 0.0 | 0.0 | 0.0 | 0.9474 | 0.8684 | 0.9211 | 0.9474 | 0.9167 | 0.8909 | 0.9333 | 0.9167 |
62
+ | 0.4243 | 3.0 | 33 | 0.2308 | 0.0 | 0.0 | 0.0 | 0.0 | 0.9474 | 0.8684 | 0.9211 | 0.9474 | 0.9167 | 0.9394 | 0.9810 | 0.9722 |
63
+ | 0.2879 | 4.0 | 44 | 0.1196 | 0.4 | 0.8889 | 0.8571 | 0.6667 | 0.9211 | 0.9737 | 0.9737 | 0.9474 | 0.9444 | 0.9939 | 1.0 | 0.9861 |
64
+ | 0.1956 | 5.0 | 55 | 0.0871 | 0.6667 | 1.0 | 0.8571 | 0.5714 | 0.9474 | 1.0 | 0.9737 | 0.9211 | 0.9583 | 1.0 | 1.0 | 1.0 |
65
+ | 0.2323 | 6.0 | 66 | 0.0961 | 0.4 | 0.9091 | 0.8571 | 0.5714 | 0.9211 | 0.9737 | 0.9737 | 0.9211 | 0.9444 | 1.0 | 1.0 | 1.0 |
66
+ | 0.1629 | 7.0 | 77 | 0.1099 | 0.4 | 1.0 | 1.0 | 0.8 | 0.9211 | 1.0 | 1.0 | 0.9737 | 0.9444 | 1.0 | 1.0 | 0.9722 |
67
+ | 0.1599 | 8.0 | 88 | 0.0819 | 0.4 | 0.7500 | 1.0 | 0.8 | 0.9211 | 0.9474 | 1.0 | 0.9737 | 0.9583 | 1.0 | 1.0 | 1.0 |
68
+ | 0.1491 | 9.0 | 99 | 0.0835 | 0.4 | 0.7500 | 1.0 | 0.8 | 0.9211 | 0.9474 | 1.0 | 0.9737 | 0.9444 | 1.0 | 1.0 | 0.9861 |
69
+ | 0.1284 | 10.0 | 110 | 0.0775 | 0.4 | 0.8889 | 1.0 | 0.8 | 0.9211 | 0.9737 | 1.0 | 0.9737 | 0.9306 | 1.0 | 1.0 | 0.9722 |
70
+ | 0.1248 | 11.0 | 121 | 0.0782 | 0.4 | 0.8889 | 1.0 | 0.8 | 0.9211 | 0.9737 | 1.0 | 0.9737 | 0.9306 | 1.0 | 1.0 | 0.9861 |
71
+ | 0.1209 | 12.0 | 132 | 0.0786 | 0.4 | 0.8889 | 1.0 | 0.8 | 0.9211 | 0.9737 | 1.0 | 0.9737 | 0.9306 | 1.0 | 1.0 | 0.9861 |
72
+ | 0.1164 | 13.0 | 143 | 0.0810 | 0.4 | 0.8889 | 1.0 | 0.8 | 0.9211 | 0.9737 | 1.0 | 0.9737 | 0.9167 | 0.9939 | 1.0 | 0.9861 |
73
+ | 0.1167 | 14.0 | 154 | 0.0813 | 0.4 | 0.8889 | 1.0 | 0.8 | 0.9211 | 0.9737 | 1.0 | 0.9737 | 0.9306 | 0.9939 | 1.0 | 0.9722 |
74
+ | 0.1216 | 15.0 | 165 | 0.0831 | 0.4 | 0.8889 | 1.0 | 0.8 | 0.9211 | 0.9737 | 1.0 | 0.9737 | 0.9306 | 0.9939 | 1.0 | 0.9861 |
75
+
76
+
77
+ ### Framework versions
78
+
79
+ - Transformers 4.36.0
80
+ - Pytorch 2.1.0+cu121
81
+ - Datasets 2.15.0
82
+ - Tokenizers 0.15.0
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "[MASK]": 128000
3
+ }
config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "austin/mimic-pubmed-deberta-small",
3
+ "architectures": [
4
+ "DebertaV2ForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.1,
9
+ "hidden_size": 768,
10
+ "id2label": {
11
+ "0": "Adventitial Inflammation",
12
+ "1": "GCA",
13
+ "2": "Giant Cells",
14
+ "3": "Intimal Hyperplasia"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
+ "label2id": {
19
+ "Adventitial Inflammation": 0,
20
+ "GCA": 1,
21
+ "Giant Cells": 2,
22
+ "Intimal Hyperplasia": 3
23
+ },
24
+ "layer_norm_eps": 1e-07,
25
+ "max_position_embeddings": 512,
26
+ "max_relative_positions": -1,
27
+ "model_type": "deberta-v2",
28
+ "norm_rel_ebd": "layer_norm",
29
+ "num_attention_heads": 12,
30
+ "num_hidden_layers": 6,
31
+ "pad_token_id": 0,
32
+ "pooler_dropout": 0,
33
+ "pooler_hidden_act": "gelu",
34
+ "pooler_hidden_size": 768,
35
+ "pos_att_type": [
36
+ "p2c",
37
+ "c2p"
38
+ ],
39
+ "position_biased_input": false,
40
+ "position_buckets": 256,
41
+ "problem_type": "multi_label_classification",
42
+ "relative_attention": true,
43
+ "share_att_key": true,
44
+ "torch_dtype": "float32",
45
+ "transformers_version": "4.36.0",
46
+ "type_vocab_size": 0,
47
+ "vocab_size": 128100
48
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd67c5c10960fea740303822e750cc7a01967f24748d1d6874f0e14a4589daaa
3
+ size 567604704
special_tokens_map.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "[CLS]",
3
+ "cls_token": "[CLS]",
4
+ "eos_token": "[SEP]",
5
+ "mask_token": "[MASK]",
6
+ "pad_token": "[PAD]",
7
+ "sep_token": "[SEP]",
8
+ "unk_token": "[UNK]"
9
+ }
spm.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c679fbf93643d19aab7ee10c0b99e460bdbc02fedf34b92b05af343b4af586fd
3
+ size 2464616
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "1": {
12
+ "content": "[CLS]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[SEP]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[UNK]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "128000": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "bos_token": "[CLS]",
45
+ "clean_up_tokenization_spaces": true,
46
+ "cls_token": "[CLS]",
47
+ "do_lower_case": false,
48
+ "eos_token": "[SEP]",
49
+ "mask_token": "[MASK]",
50
+ "model_max_length": 1000000000000000019884624838656,
51
+ "pad_token": "[PAD]",
52
+ "sep_token": "[SEP]",
53
+ "sp_model_kwargs": {},
54
+ "split_by_punct": false,
55
+ "tokenizer_class": "DebertaV2Tokenizer",
56
+ "unk_token": "[UNK]",
57
+ "vocab_type": "spm"
58
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:866007dfe9a0a689f553d5d01d09bd80f561a6c398c15169808ba32edd955c2c
3
+ size 4728