Mikask commited on
Commit
292e40e
·
verified ·
1 Parent(s): ac85b2f

End of training

Browse files
Files changed (4) hide show
  1. README.md +69 -0
  2. config.json +86 -0
  3. pytorch_model.bin +3 -0
  4. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - generated_from_trainer
4
+ metrics:
5
+ - accuracy
6
+ - precision
7
+ - recall
8
+ - f1
9
+ model-index:
10
+ - name: bdc2024-indobert-2
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # bdc2024-indobert-2
18
+
19
+ This model was trained from scratch on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.5554
22
+ - Accuracy: 0.9331
23
+ - Balanced Accuracy: 0.8724
24
+ - Precision: 0.9353
25
+ - Recall: 0.9331
26
+ - F1: 0.9289
27
+
28
+ ## Model description
29
+
30
+ More information needed
31
+
32
+ ## Intended uses & limitations
33
+
34
+ More information needed
35
+
36
+ ## Training and evaluation data
37
+
38
+ More information needed
39
+
40
+ ## Training procedure
41
+
42
+ ### Training hyperparameters
43
+
44
+ The following hyperparameters were used during training:
45
+ - learning_rate: 1e-05
46
+ - train_batch_size: 16
47
+ - eval_batch_size: 16
48
+ - seed: 42
49
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
+ - lr_scheduler_type: linear
51
+ - num_epochs: 5
52
+
53
+ ### Training results
54
+
55
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Balanced Accuracy | Precision | Recall | F1 |
56
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:-----------------:|:---------:|:------:|:------:|
57
+ | No log | 1.0 | 483 | 0.5108 | 0.9197 | 0.8454 | 0.9195 | 0.9197 | 0.9132 |
58
+ | 0.0773 | 2.0 | 966 | 0.5374 | 0.9235 | 0.8668 | 0.9266 | 0.9235 | 0.9196 |
59
+ | 0.0374 | 3.0 | 1449 | 0.5451 | 0.9331 | 0.8689 | 0.9359 | 0.9331 | 0.9281 |
60
+ | 0.0242 | 4.0 | 1932 | 0.5567 | 0.9331 | 0.8726 | 0.9353 | 0.9331 | 0.9288 |
61
+ | 0.0162 | 5.0 | 2415 | 0.5554 | 0.9331 | 0.8724 | 0.9353 | 0.9331 | 0.9289 |
62
+
63
+
64
+ ### Framework versions
65
+
66
+ - Transformers 4.33.1
67
+ - Pytorch 2.3.0+cu121
68
+ - Datasets 2.19.2
69
+ - Tokenizers 0.13.3
config.json ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "Mikask/bdc2024-indobert-1",
3
+ "activation_dropout": 0.1,
4
+ "activation_function": "gelu",
5
+ "add_bias_logits": false,
6
+ "add_final_layer_norm": false,
7
+ "architectures": [
8
+ "MBartForSequenceClassification"
9
+ ],
10
+ "attention_dropout": 0.1,
11
+ "bos_token_id": 0,
12
+ "classif_dropout": 0.1,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 768,
15
+ "decoder_attention_heads": 12,
16
+ "decoder_ffn_dim": 3072,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 6,
19
+ "decoder_start_token_id": 2,
20
+ "dropout": 0.1,
21
+ "early_stopping": true,
22
+ "encoder_attention_heads": 12,
23
+ "encoder_ffn_dim": 3072,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 6,
26
+ "eos_token_id": 2,
27
+ "forced_eos_token_id": 2,
28
+ "gradient_checkpointing": false,
29
+ "id2label": {
30
+ "0": "LABEL_0",
31
+ "1": "LABEL_1",
32
+ "2": "LABEL_2",
33
+ "3": "LABEL_3",
34
+ "4": "LABEL_4",
35
+ "5": "LABEL_5",
36
+ "6": "LABEL_6",
37
+ "7": "LABEL_7"
38
+ },
39
+ "init_std": 0.02,
40
+ "is_encoder_decoder": true,
41
+ "label2id": {
42
+ "LABEL_0": 0,
43
+ "LABEL_1": 1,
44
+ "LABEL_2": 2,
45
+ "LABEL_3": 3,
46
+ "LABEL_4": 4,
47
+ "LABEL_5": 5,
48
+ "LABEL_6": 6,
49
+ "LABEL_7": 7
50
+ },
51
+ "max_position_embeddings": 1024,
52
+ "model_type": "mbart",
53
+ "no_repeat_ngram_size": 3,
54
+ "normalize_before": false,
55
+ "normalize_embedding": true,
56
+ "num_beams": 4,
57
+ "num_hidden_layers": 6,
58
+ "pad_token_id": 1,
59
+ "problem_type": "single_label_classification",
60
+ "scale_embedding": false,
61
+ "task_specific_params": {
62
+ "summarization": {
63
+ "length_penalty": 1.0,
64
+ "max_length": 128,
65
+ "min_length": 12,
66
+ "num_beams": 4
67
+ },
68
+ "summarization_cnn": {
69
+ "length_penalty": 2.0,
70
+ "max_length": 142,
71
+ "min_length": 56,
72
+ "num_beams": 4
73
+ },
74
+ "summarization_xsum": {
75
+ "length_penalty": 1.0,
76
+ "max_length": 62,
77
+ "min_length": 11,
78
+ "num_beams": 6
79
+ }
80
+ },
81
+ "tokenizer_class": "IndoNLGTokenizer",
82
+ "torch_dtype": "float32",
83
+ "transformers_version": "4.33.1",
84
+ "use_cache": true,
85
+ "vocab_size": 40004
86
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:226b80e2f0c5d638abb2c49d03cab0bd0cddad6241e10530e39011b07113da9e
3
+ size 528650230
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:862326d32c49ad6e3b6a977349f745ca2a4dd515fdd30ad864ab56eb8ef9a8b9
3
+ size 4536