LeoZZzzZZ commited on
Commit
8906d70
1 Parent(s): 100993b

Training in progress epoch 0

Browse files
README.md CHANGED
@@ -15,13 +15,13 @@ probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.2943
19
- - Validation Loss: 0.5154
20
- - Train Accuracy: 0.8145
21
- - Train Precision: 0.8130
22
- - Train Recall: 0.8145
23
- - Train F1: 0.8123
24
- - Epoch: 2
25
 
26
  ## Model description
27
 
@@ -40,21 +40,19 @@ More information needed
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
- - optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 18780, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False}
44
  - training_precision: float32
45
 
46
  ### Training results
47
 
48
  | Train Loss | Validation Loss | Train Accuracy | Train Precision | Train Recall | Train F1 | Epoch |
49
  |:----------:|:---------------:|:--------------:|:---------------:|:------------:|:--------:|:-----:|
50
- | 0.7207 | 0.5929 | 0.7559 | 0.7512 | 0.7559 | 0.7520 | 0 |
51
- | 0.4594 | 0.4977 | 0.8052 | 0.8043 | 0.8052 | 0.8046 | 1 |
52
- | 0.2943 | 0.5154 | 0.8145 | 0.8130 | 0.8145 | 0.8123 | 2 |
53
 
54
 
55
  ### Framework versions
56
 
57
- - Transformers 4.40.2
58
  - TensorFlow 2.15.0
59
  - Datasets 2.19.1
60
  - Tokenizers 0.19.1
 
15
 
16
  This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 0.5459
19
+ - Validation Loss: 0.7333
20
+ - Train Accuracy: 0.6880
21
+ - Train Precision: 0.7015
22
+ - Train Recall: 0.6880
23
+ - Train F1: 0.6863
24
+ - Epoch: 0
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 90930, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False}
44
  - training_precision: float32
45
 
46
  ### Training results
47
 
48
  | Train Loss | Validation Loss | Train Accuracy | Train Precision | Train Recall | Train F1 | Epoch |
49
  |:----------:|:---------------:|:--------------:|:---------------:|:------------:|:--------:|:-----:|
50
+ | 0.5459 | 0.7333 | 0.6880 | 0.7015 | 0.6880 | 0.6863 | 0 |
 
 
51
 
52
 
53
  ### Framework versions
54
 
55
+ - Transformers 4.41.0
56
  - TensorFlow 2.15.0
57
  - Datasets 2.19.1
58
  - Tokenizers 0.19.1
config.json CHANGED
@@ -10,16 +10,16 @@
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
- "0": "NOT ENOUGH INFO",
14
- "1": "SUPPORTS",
15
- "2": "REFUTES"
16
  },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
- "NOT ENOUGH INFO": 0,
21
- "REFUTES": 2,
22
- "SUPPORTS": 1
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "max_position_embeddings": 512,
@@ -29,7 +29,7 @@
29
  "pad_token_id": 0,
30
  "position_embedding_type": "absolute",
31
  "problem_type": "multi_label_classification",
32
- "transformers_version": "4.40.2",
33
  "type_vocab_size": 2,
34
  "use_cache": true,
35
  "vocab_size": 30522
 
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
+ "0": "SUPPORTS",
14
+ "1": "REFUTES",
15
+ "2": "NOT ENOUGH INFO"
16
  },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
+ "NOT ENOUGH INFO": 2,
21
+ "REFUTES": 1,
22
+ "SUPPORTS": 0
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "max_position_embeddings": 512,
 
29
  "pad_token_id": 0,
30
  "position_embedding_type": "absolute",
31
  "problem_type": "multi_label_classification",
32
+ "transformers_version": "4.41.0",
33
  "type_vocab_size": 2,
34
  "use_cache": true,
35
  "vocab_size": 30522
logs/train/events.out.tfevents.1716351588.935214a22de4.5785.0.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7c18da95085660382eee0160a45b730c19005b862275d6bc12cdb1cd4666ad5
3
+ size 3745779
logs/validation/events.out.tfevents.1716352186.935214a22de4.5785.1.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cbb7a7308b4296966a6a73a6ed7401541181b7100689c3ab604500f320c0a9c
3
+ size 232
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e9331e087361c3fd76653bd6e34b39b03201a381bb9b1a275ddd40ffb7fc71f1
3
  size 438226204
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de6f1be4bbcd3cc419202363bd89fd2187f615a3f63110560783523d0c3a6eb9
3
  size 438226204