MaryaAI commited on
Commit
c05fd7d
1 Parent(s): 1c38b63
Files changed (3) hide show
  1. README.md +68 -0
  2. config.json +58 -0
  3. tf_model.h5 +3 -0
README.md ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_keras_callback
5
+ model-index:
6
+ - name: opus-mt-ar-en-finetunedTanzil-v6-ar-to-en
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information Keras had access to. You should
11
+ probably proofread and complete it, then remove this comment. -->
12
+
13
+ # opus-mt-ar-en-finetunedTanzil-v6-ar-to-en
14
+
15
+ This model is a fine-tuned version of [Helsinki-NLP/opus-mt-ar-en](https://huggingface.co/Helsinki-NLP/opus-mt-ar-en) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+ - Train Loss: 0.8405
18
+ - Validation Loss: 0.9594
19
+ - Train Rouge1: 58.4345
20
+ - Train Rouge2: 34.4502
21
+ - Train Rougel: 54.6634
22
+ - Train Rougelsum: 54.9469
23
+ - Train Gen Len: 96.98
24
+ - Epoch: 10
25
+
26
+ ## Model description
27
+
28
+ More information needed
29
+
30
+ ## Intended uses & limitations
31
+
32
+ More information needed
33
+
34
+ ## Training and evaluation data
35
+
36
+ More information needed
37
+
38
+ ## Training procedure
39
+
40
+ ### Training hyperparameters
41
+
42
+ The following hyperparameters were used during training:
43
+ - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False, 'weight_decay_rate': 0.01}
44
+ - training_precision: float32
45
+
46
+ ### Training results
47
+
48
+ | Train Loss | Validation Loss | Train Rouge1 | Train Rouge2 | Train Rougel | Train Rougelsum | Train Gen Len | Epoch |
49
+ |:----------:|:---------------:|:------------:|:------------:|:------------:|:---------------:|:-------------:|:-----:|
50
+ | 1.7341 | 1.4426 | 55.0505 | 31.6295 | 51.3925 | 51.8558 | 76.44 | 0 |
51
+ | 1.4329 | 1.2987 | 56.7672 | 33.8519 | 53.2796 | 53.6077 | 75.905 | 1 |
52
+ | 1.2849 | 1.2132 | 59.1926 | 36.8577 | 55.8615 | 56.2321 | 94.415 | 2 |
53
+ | 1.1854 | 1.1560 | 59.5654 | 36.0992 | 56.2438 | 56.6259 | 109.725 | 3 |
54
+ | 1.1092 | 1.1101 | 59.2882 | 36.3247 | 56.1751 | 56.6098 | 110.7 | 4 |
55
+ | 1.0478 | 1.0738 | 59.3702 | 35.9485 | 56.2157 | 56.6418 | 100.835 | 5 |
56
+ | 0.9957 | 1.0464 | 59.3582 | 35.5831 | 55.8989 | 56.3337 | 103.575 | 6 |
57
+ | 0.9503 | 1.0190 | 59.2327 | 35.7054 | 55.6769 | 56.0504 | 97.95 | 7 |
58
+ | 0.9099 | 0.9959 | 58.9992 | 34.9965 | 55.5582 | 55.8486 | 97.05 | 8 |
59
+ | 0.8735 | 0.9776 | 58.9665 | 35.1024 | 55.6433 | 55.8640 | 100.195 | 9 |
60
+ | 0.8405 | 0.9594 | 58.4345 | 34.4502 | 54.6634 | 54.9469 | 96.98 | 10 |
61
+
62
+
63
+ ### Framework versions
64
+
65
+ - Transformers 4.17.0.dev0
66
+ - TensorFlow 2.7.0
67
+ - Datasets 1.18.4.dev0
68
+ - Tokenizers 0.10.3
config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "Helsinki-NLP/opus-mt-ar-en",
3
+ "_num_labels": 3,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "swish",
6
+ "add_bias_logits": false,
7
+ "add_final_layer_norm": false,
8
+ "architectures": [
9
+ "MarianMTModel"
10
+ ],
11
+ "attention_dropout": 0.0,
12
+ "bad_words_ids": [
13
+ [
14
+ 62833
15
+ ]
16
+ ],
17
+ "bos_token_id": 0,
18
+ "classif_dropout": 0.0,
19
+ "classifier_dropout": 0.0,
20
+ "d_model": 512,
21
+ "decoder_attention_heads": 8,
22
+ "decoder_ffn_dim": 2048,
23
+ "decoder_layerdrop": 0.0,
24
+ "decoder_layers": 6,
25
+ "decoder_start_token_id": 62833,
26
+ "dropout": 0.1,
27
+ "encoder_attention_heads": 8,
28
+ "encoder_ffn_dim": 2048,
29
+ "encoder_layerdrop": 0.0,
30
+ "encoder_layers": 6,
31
+ "eos_token_id": 0,
32
+ "forced_eos_token_id": 0,
33
+ "id2label": {
34
+ "0": "LABEL_0",
35
+ "1": "LABEL_1",
36
+ "2": "LABEL_2"
37
+ },
38
+ "init_std": 0.02,
39
+ "is_encoder_decoder": true,
40
+ "label2id": {
41
+ "LABEL_0": 0,
42
+ "LABEL_1": 1,
43
+ "LABEL_2": 2
44
+ },
45
+ "max_length": 512,
46
+ "max_position_embeddings": 512,
47
+ "model_type": "marian",
48
+ "normalize_before": false,
49
+ "normalize_embedding": false,
50
+ "num_beams": 4,
51
+ "num_hidden_layers": 6,
52
+ "pad_token_id": 62833,
53
+ "scale_embedding": true,
54
+ "static_position_embeddings": true,
55
+ "transformers_version": "4.17.0.dev0",
56
+ "use_cache": true,
57
+ "vocab_size": 62834
58
+ }
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c19e1e49298b8af0992b7c32715333deddba68431ee4ef674ec1fdc03bc111e
3
+ size 307854560