eloi-goncalves commited on
Commit
8d42f28
1 Parent(s): c576f56

Training in progress epoch 0

Browse files
Files changed (3) hide show
  1. README.md +6 -8
  2. config.json +53 -9
  3. tf_model.h5 +2 -2
README.md CHANGED
@@ -15,9 +15,9 @@ probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [neuralmind/bert-base-portuguese-cased](https://huggingface.co/neuralmind/bert-base-portuguese-cased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.3991
19
- - Validation Loss: 0.3952
20
- - Epoch: 2
21
 
22
  ## Model description
23
 
@@ -36,21 +36,19 @@ More information needed
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
- - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 624, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
40
  - training_precision: mixed_float16
41
 
42
  ### Training results
43
 
44
  | Train Loss | Validation Loss | Epoch |
45
  |:----------:|:---------------:|:-----:|
46
- | 0.5389 | 0.4231 | 0 |
47
- | 0.4175 | 0.3972 | 1 |
48
- | 0.3991 | 0.3952 | 2 |
49
 
50
 
51
  ### Framework versions
52
 
53
- - Transformers 4.34.1
54
  - TensorFlow 2.14.0
55
  - Datasets 2.14.6
56
  - Tokenizers 0.14.1
 
15
 
16
  This model is a fine-tuned version of [neuralmind/bert-base-portuguese-cased](https://huggingface.co/neuralmind/bert-base-portuguese-cased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 0.0701
19
+ - Validation Loss: 0.0060
20
+ - Epoch: 0
21
 
22
  ## Model description
23
 
 
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
+ - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 7647, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
40
  - training_precision: mixed_float16
41
 
42
  ### Training results
43
 
44
  | Train Loss | Validation Loss | Epoch |
45
  |:----------:|:---------------:|:-----:|
46
+ | 0.0701 | 0.0060 | 0 |
 
 
47
 
48
 
49
  ### Framework versions
50
 
51
+ - Transformers 4.35.0
52
  - TensorFlow 2.14.0
53
  - Datasets 2.14.6
54
  - Tokenizers 0.14.1
config.json CHANGED
@@ -11,18 +11,62 @@
11
  "hidden_size": 768,
12
  "id2label": {
13
  "0": "O",
14
- "1": "B-SUBMIT_CONTEXT",
15
- "2": "I-SUBMIT_CONTEXT",
16
- "3": "B-PATIENT_NAME",
17
- "4": "I-PATIENT_NAME"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
  },
19
  "initializer_range": 0.02,
20
  "intermediate_size": 3072,
21
  "label2id": {
22
- "B-PATIENT_NAME": 3,
23
- "B-SUBMIT_CONTEXT": 1,
24
- "I-PATIENT_NAME": 4,
25
- "I-SUBMIT_CONTEXT": 2,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
26
  "O": 0
27
  },
28
  "layer_norm_eps": 1e-12,
@@ -38,7 +82,7 @@
38
  "pooler_size_per_head": 128,
39
  "pooler_type": "first_token_transform",
40
  "position_embedding_type": "absolute",
41
- "transformers_version": "4.34.1",
42
  "type_vocab_size": 2,
43
  "use_cache": true,
44
  "vocab_size": 29794
 
11
  "hidden_size": 768,
12
  "id2label": {
13
  "0": "O",
14
+ "1": "B-MACRO_NAME",
15
+ "2": "I-MACRO_NAME",
16
+ "3": "B-ROUTINE_NAME",
17
+ "4": "I-ROUTINE_NAME",
18
+ "5": "B-SHOWME_SPECIFIC_VITALS_MODAL",
19
+ "6": "I-SHOWME_SPECIFIC_VITALS_MODAL",
20
+ "7": "B-SUBMIT_CONTEXT",
21
+ "8": "I-SUBMIT_CONTEXT",
22
+ "9": "B-UNIT_NAME",
23
+ "10": "I-UNIT_NAME",
24
+ "11": "B-GO_TO_LOCATION",
25
+ "12": "I-GO_TO_LOCATION",
26
+ "13": "B-ORDER_ITEM",
27
+ "14": "I-ORDER_ITEM",
28
+ "15": "B-PATIENT_NAME",
29
+ "16": "I-PATIENT_NAME",
30
+ "17": "B-CLINICAL_NOTE",
31
+ "18": "I-CLINICAL_NOTE",
32
+ "19": "B-TEMPLATE_NAME",
33
+ "20": "I-TEMPLATE_NAME",
34
+ "21": "B-SHOWME_SPECIFIC_LABRESULTS_MODAL",
35
+ "22": "I-SHOWME_SPECIFIC_LABRESULTS_MODAL",
36
+ "23": "B-SHOWME_SPECIFIC_SCORE_MODAL",
37
+ "24": "I-SHOWME_SPECIFIC_SCORE_MODAL",
38
+ "25": "B-PROTOCOL_NAME",
39
+ "26": "I-PROTOCOL_NAME"
40
  },
41
  "initializer_range": 0.02,
42
  "intermediate_size": 3072,
43
  "label2id": {
44
+ "B-CLINICAL_NOTE": 17,
45
+ "B-GO_TO_LOCATION": 11,
46
+ "B-MACRO_NAME": 1,
47
+ "B-ORDER_ITEM": 13,
48
+ "B-PATIENT_NAME": 15,
49
+ "B-PROTOCOL_NAME": 25,
50
+ "B-ROUTINE_NAME": 3,
51
+ "B-SHOWME_SPECIFIC_LABRESULTS_MODAL": 21,
52
+ "B-SHOWME_SPECIFIC_SCORE_MODAL": 23,
53
+ "B-SHOWME_SPECIFIC_VITALS_MODAL": 5,
54
+ "B-SUBMIT_CONTEXT": 7,
55
+ "B-TEMPLATE_NAME": 19,
56
+ "B-UNIT_NAME": 9,
57
+ "I-CLINICAL_NOTE": 18,
58
+ "I-GO_TO_LOCATION": 12,
59
+ "I-MACRO_NAME": 2,
60
+ "I-ORDER_ITEM": 14,
61
+ "I-PATIENT_NAME": 16,
62
+ "I-PROTOCOL_NAME": 26,
63
+ "I-ROUTINE_NAME": 4,
64
+ "I-SHOWME_SPECIFIC_LABRESULTS_MODAL": 22,
65
+ "I-SHOWME_SPECIFIC_SCORE_MODAL": 24,
66
+ "I-SHOWME_SPECIFIC_VITALS_MODAL": 6,
67
+ "I-SUBMIT_CONTEXT": 8,
68
+ "I-TEMPLATE_NAME": 20,
69
+ "I-UNIT_NAME": 10,
70
  "O": 0
71
  },
72
  "layer_norm_eps": 1e-12,
 
82
  "pooler_size_per_head": 128,
83
  "pooler_type": "first_token_transform",
84
  "position_embedding_type": "absolute",
85
+ "transformers_version": "4.35.0",
86
  "type_vocab_size": 2,
87
  "use_cache": true,
88
  "vocab_size": 29794
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8a243d2f96f73bc2e6ddf9b714a5311df9445be266455b01ddf8515df05a525a
3
- size 433618908
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:625a23140660b729ecdcb24ab3a088cb61de1d4e886b9cfec2ea373020b634e6
3
+ size 433686580