Flamgrise commited on
Commit
423adc6
1 Parent(s): 94d0d79

Fine-tuned BART model for League of Legends biographies classification

Browse files
Files changed (24) hide show
  1. README.md +55 -0
  2. config.json +58 -0
  3. merges.txt +0 -0
  4. model.safetensors +3 -0
  5. runs/Mar22_09-54-02_0c73de3e84dd/events.out.tfevents.1711101333.0c73de3e84dd.312.0 +3 -0
  6. runs/Mar22_10-01-31_0c73de3e84dd/events.out.tfevents.1711101697.0c73de3e84dd.312.1 +3 -0
  7. runs/Mar22_10-41-03_0c73de3e84dd/events.out.tfevents.1711104070.0c73de3e84dd.312.2 +3 -0
  8. runs/Mar22_11-17-17_0c73de3e84dd/events.out.tfevents.1711106242.0c73de3e84dd.312.3 +3 -0
  9. runs/Mar22_12-06-10_0c73de3e84dd/events.out.tfevents.1711109192.0c73de3e84dd.312.4 +3 -0
  10. runs/Mar22_12-49-06_0c73de3e84dd/events.out.tfevents.1711111751.0c73de3e84dd.312.5 +3 -0
  11. runs/Mar22_12-50-27_0c73de3e84dd/events.out.tfevents.1711111833.0c73de3e84dd.312.6 +3 -0
  12. runs/Mar22_12-57-10_0c73de3e84dd/events.out.tfevents.1711112237.0c73de3e84dd.45184.0 +3 -0
  13. runs/Mar22_12-57-10_0c73de3e84dd/events.out.tfevents.1711114533.0c73de3e84dd.45184.1 +3 -0
  14. runs/Mar22_13-41-33_0c73de3e84dd/events.out.tfevents.1711114899.0c73de3e84dd.45184.2 +3 -0
  15. runs/Mar22_13-42-32_0c73de3e84dd/events.out.tfevents.1711114958.0c73de3e84dd.45184.3 +3 -0
  16. runs/Mar22_14-18-11_0c73de3e84dd/events.out.tfevents.1711117097.0c73de3e84dd.45184.4 +3 -0
  17. runs/Mar22_14-54-37_0c73de3e84dd/events.out.tfevents.1711119282.0c73de3e84dd.45184.5 +3 -0
  18. runs/Mar22_14-54-37_0c73de3e84dd/events.out.tfevents.1711119311.0c73de3e84dd.45184.6 +3 -0
  19. runs/Mar22_14-54-37_0c73de3e84dd/events.out.tfevents.1711119322.0c73de3e84dd.45184.7 +3 -0
  20. special_tokens_map.json +15 -0
  21. tokenizer.json +0 -0
  22. tokenizer_config.json +57 -0
  23. training_args.bin +3 -0
  24. vocab.json +0 -0
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ base_model: facebook/bart-large-mnli
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: bios_lol_fine-tuned
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # bios_lol_fine-tuned
15
+
16
+ This model is a fine-tuned version of [facebook/bart-large-mnli](https://huggingface.co/facebook/bart-large-mnli) on an unknown dataset.
17
+
18
+ ## Model description
19
+
20
+ More information needed
21
+
22
+ ## Intended uses & limitations
23
+
24
+ More information needed
25
+
26
+ ## Training and evaluation data
27
+
28
+ More information needed
29
+
30
+ ## Training procedure
31
+
32
+ ### Training hyperparameters
33
+
34
+ The following hyperparameters were used during training:
35
+ - learning_rate: 2e-05
36
+ - train_batch_size: 1
37
+ - eval_batch_size: 1
38
+ - seed: 42
39
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
40
+ - lr_scheduler_type: linear
41
+ - num_epochs: 1
42
+
43
+ ### Training results
44
+
45
+ | Training Loss | Epoch | Step | Validation Loss | F1 |
46
+ |:-------------:|:-----:|:----:|:---------------:|:------:|
47
+ | No log | 1.0 | 130 | 1.7029 | 0.0816 |
48
+
49
+
50
+ ### Framework versions
51
+
52
+ - Transformers 4.38.2
53
+ - Pytorch 2.2.1+cu121
54
+ - Datasets 2.18.0
55
+ - Tokenizers 0.15.2
config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/bart-large-mnli",
3
+ "_num_labels": 3,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "gelu",
6
+ "add_final_layer_norm": false,
7
+ "architectures": [
8
+ "BartForSequenceClassification"
9
+ ],
10
+ "attention_dropout": 0.0,
11
+ "bos_token_id": 0,
12
+ "classif_dropout": 0.0,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 1024,
15
+ "decoder_attention_heads": 16,
16
+ "decoder_ffn_dim": 4096,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 12,
19
+ "decoder_start_token_id": 2,
20
+ "dropout": 0.1,
21
+ "encoder_attention_heads": 16,
22
+ "encoder_ffn_dim": 4096,
23
+ "encoder_layerdrop": 0.0,
24
+ "encoder_layers": 12,
25
+ "eos_token_id": 2,
26
+ "forced_eos_token_id": 2,
27
+ "gradient_checkpointing": false,
28
+ "id2label": {
29
+ "0": "Fighter",
30
+ "1": "Mage",
31
+ "2": "Assassin",
32
+ "3": "Marksman",
33
+ "4": "Tank",
34
+ "5": "Support"
35
+ },
36
+ "init_std": 0.02,
37
+ "is_encoder_decoder": true,
38
+ "label2id": {
39
+ "Assassin": 2,
40
+ "Fighter": 0,
41
+ "Mage": 1,
42
+ "Marksman": 3,
43
+ "Support": 5,
44
+ "Tank": 4
45
+ },
46
+ "max_position_embeddings": 1024,
47
+ "model_type": "bart",
48
+ "normalize_before": false,
49
+ "num_hidden_layers": 12,
50
+ "output_past": false,
51
+ "pad_token_id": 1,
52
+ "problem_type": "single_label_classification",
53
+ "scale_embedding": false,
54
+ "torch_dtype": "float32",
55
+ "transformers_version": "4.38.2",
56
+ "use_cache": true,
57
+ "vocab_size": 50265
58
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac418bf396ffe41932fcdef7ffa125a3e6bfa3cc118750943a09b3a2f3545f77
3
+ size 1629449264
runs/Mar22_09-54-02_0c73de3e84dd/events.out.tfevents.1711101333.0c73de3e84dd.312.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1434f5f269bc0152c76b66960f14c64dbb3899f374fb8e91cff8028f4344791f
3
+ size 5213
runs/Mar22_10-01-31_0c73de3e84dd/events.out.tfevents.1711101697.0c73de3e84dd.312.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ddcfce116d0af7688104c00f8a525f0cba794c87866c74f110e01e8cdf007b2a
3
+ size 5262
runs/Mar22_10-41-03_0c73de3e84dd/events.out.tfevents.1711104070.0c73de3e84dd.312.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c92b7fb48195454e62851cfa4b203445fdd4d1158ca9fdc013bf97cf522bf98c
3
+ size 5213
runs/Mar22_11-17-17_0c73de3e84dd/events.out.tfevents.1711106242.0c73de3e84dd.312.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea329caaac537bd856049ccd35cc2b168ff36d4d7078a5fcaaba70d86a004001
3
+ size 5213
runs/Mar22_12-06-10_0c73de3e84dd/events.out.tfevents.1711109192.0c73de3e84dd.312.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:012fe3b628b4a3a1c05b4a364ea73063798c6ab2ea67b986917dc55bdc5ff391
3
+ size 10452
runs/Mar22_12-49-06_0c73de3e84dd/events.out.tfevents.1711111751.0c73de3e84dd.312.5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f14327edf8761aeaaf0066d5983fd350003b9d52d5fc28213da7e75ebd03738
3
+ size 5213
runs/Mar22_12-50-27_0c73de3e84dd/events.out.tfevents.1711111833.0c73de3e84dd.312.6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3be6cfcd508f093d8237f1489b25fb50b13c5d6a13ba9f2bb6c6a6d510799172
3
+ size 5213
runs/Mar22_12-57-10_0c73de3e84dd/events.out.tfevents.1711112237.0c73de3e84dd.45184.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:784dae6f1e4b26419a4b46e16544d37917a52e2963f526a615e1962ae9ad1c47
3
+ size 5214
runs/Mar22_12-57-10_0c73de3e84dd/events.out.tfevents.1711114533.0c73de3e84dd.45184.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b61ffc1b279b9ce66057ab7d4f2c615f1c772b4786710705cb620655465cea0
3
+ size 5126
runs/Mar22_13-41-33_0c73de3e84dd/events.out.tfevents.1711114899.0c73de3e84dd.45184.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f36aed5082e34760a4c3b5ab2cd287211582733e94bcf2b3088a8c77f89ac4e4
3
+ size 5126
runs/Mar22_13-42-32_0c73de3e84dd/events.out.tfevents.1711114958.0c73de3e84dd.45184.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8468adaec1907a01cbb429ddb551bf4a4246bc7ecbe9dd35694fd23e5a1e0ba
3
+ size 5214
runs/Mar22_14-18-11_0c73de3e84dd/events.out.tfevents.1711117097.0c73de3e84dd.45184.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:acd143ed05d39596ff87c6a8951e2ce054521c5233f51f263693f94a481786ef
3
+ size 5214
runs/Mar22_14-54-37_0c73de3e84dd/events.out.tfevents.1711119282.0c73de3e84dd.45184.5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4bf2b79d5fa4d32fefcc27c550c04a201a7e6dc8b681166b576f26301efe116
3
+ size 5263
runs/Mar22_14-54-37_0c73de3e84dd/events.out.tfevents.1711119311.0c73de3e84dd.45184.6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fad404572f7aa9353e57268aa0c6cc120be8f0c8ba88433c82dca2d44febd31f
3
+ size 5263
runs/Mar22_14-54-37_0c73de3e84dd/events.out.tfevents.1711119322.0c73de3e84dd.45184.7 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9fe3ec7020b703df49a877437a963719cd08fa2ee605287875f3c15b44a05d3
3
+ size 5934
special_tokens_map.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "cls_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "mask_token": {
6
+ "content": "<mask>",
7
+ "lstrip": true,
8
+ "normalized": true,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "pad_token": "<pad>",
13
+ "sep_token": "</s>",
14
+ "unk_token": "<unk>"
15
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": true,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": true,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "mask_token": "<mask>",
51
+ "model_max_length": 1024,
52
+ "pad_token": "<pad>",
53
+ "sep_token": "</s>",
54
+ "tokenizer_class": "BartTokenizer",
55
+ "trim_offsets": true,
56
+ "unk_token": "<unk>"
57
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:899bb9a8f727122ab18757d730ffdd358397c6bafe3131e3bf8cf5d7b2a3acc1
3
+ size 4920
vocab.json ADDED
The diff for this file is too large to render. See raw diff