Himanshu99001 commited on
Commit
3640c14
·
verified ·
1 Parent(s): 1b0d2d8

Initial model upload

Browse files
README.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: albert/albert-base-v2
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ - f1
10
+ - precision
11
+ - recall
12
+ model-index:
13
+ - name: classify-bluesky-1000-v2
14
+ results: []
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+ # classify-bluesky-1000-v2
21
+
22
+ This model is a fine-tuned version of [albert/albert-base-v2](https://huggingface.co/albert/albert-base-v2) on an unknown dataset.
23
+ It achieves the following results on the evaluation set:
24
+ - Loss: 0.1265
25
+ - Accuracy: 1.0
26
+ - F1: 1.0
27
+ - Precision: 1.0
28
+ - Recall: 1.0
29
+ - Accuracy Label Bluesky: 1.0
30
+ - Accuracy Label Specific: 1.0
31
+
32
+ ## Model description
33
+
34
+ More information needed
35
+
36
+ ## Intended uses & limitations
37
+
38
+ More information needed
39
+
40
+ ## Training and evaluation data
41
+
42
+ More information needed
43
+
44
+ ## Training procedure
45
+
46
+ ### Training hyperparameters
47
+
48
+ The following hyperparameters were used during training:
49
+ - learning_rate: 2e-05
50
+ - train_batch_size: 16
51
+ - eval_batch_size: 16
52
+ - seed: 42
53
+ - gradient_accumulation_steps: 2
54
+ - total_train_batch_size: 32
55
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
56
+ - lr_scheduler_type: linear
57
+ - lr_scheduler_warmup_steps: 500
58
+ - num_epochs: 3
59
+
60
+ ### Training results
61
+
62
+
63
+
64
+ ### Framework versions
65
+
66
+ - Transformers 4.45.2
67
+ - Pytorch 2.4.1+cu121
68
+ - Datasets 3.0.1
69
+ - Tokenizers 0.20.1
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "albert/albert-base-v2",
3
+ "architectures": [
4
+ "AlbertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0,
7
+ "bos_token_id": 2,
8
+ "classifier_dropout_prob": 0.1,
9
+ "down_scale_factor": 1,
10
+ "embedding_size": 128,
11
+ "eos_token_id": 3,
12
+ "gap_size": 0,
13
+ "hidden_act": "gelu_new",
14
+ "hidden_dropout_prob": 0,
15
+ "hidden_size": 768,
16
+ "id2label": {
17
+ "0": "bluesky",
18
+ "1": "specific"
19
+ },
20
+ "initializer_range": 0.02,
21
+ "inner_group_num": 1,
22
+ "intermediate_size": 3072,
23
+ "label2id": {
24
+ "bluesky": 0,
25
+ "specific": 1
26
+ },
27
+ "layer_norm_eps": 1e-12,
28
+ "max_position_embeddings": 512,
29
+ "model_type": "albert",
30
+ "net_structure_type": 0,
31
+ "num_attention_heads": 12,
32
+ "num_hidden_groups": 1,
33
+ "num_hidden_layers": 12,
34
+ "num_memory_blocks": 0,
35
+ "pad_token_id": 0,
36
+ "position_embedding_type": "absolute",
37
+ "problem_type": "single_label_classification",
38
+ "torch_dtype": "float32",
39
+ "transformers_version": "4.45.2",
40
+ "type_vocab_size": 2,
41
+ "vocab_size": 30000
42
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0198cfa34374703bbbd83d4f3575d6364d1183319cc0ca2edde4e9b75664e22
3
+ size 46743912
runs/Oct15_06-15-58_3a6228660927/events.out.tfevents.1728973095.3a6228660927.1914.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20947d5ac625a1aa3cc428d64f9ee183e6a23cd35cc09e855eb0b97506ba1654
3
+ size 6829
runs/Oct15_06-15-58_3a6228660927/events.out.tfevents.1728973124.3a6228660927.1914.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a33dc46d961ce4581655a9ab3063442e55bd1c78b7ddbbddb9a17b1fe6d7013d
3
+ size 682
special_tokens_map.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "[CLS]",
3
+ "cls_token": "[CLS]",
4
+ "eos_token": "[SEP]",
5
+ "mask_token": {
6
+ "content": "[MASK]",
7
+ "lstrip": true,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "pad_token": "<pad>",
13
+ "sep_token": "[SEP]",
14
+ "unk_token": "<unk>"
15
+ }
spiece.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fefb02b667a6c5c2fe27602d28e5fb3428f66ab89c7d6f388e7c8d44a02d0336
3
+ size 760289
tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<pad>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<unk>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "[MASK]",
37
+ "lstrip": true,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "bos_token": "[CLS]",
45
+ "clean_up_tokenization_spaces": false,
46
+ "cls_token": "[CLS]",
47
+ "do_lower_case": true,
48
+ "eos_token": "[SEP]",
49
+ "keep_accents": false,
50
+ "mask_token": "[MASK]",
51
+ "model_max_length": 512,
52
+ "pad_token": "<pad>",
53
+ "remove_space": true,
54
+ "sep_token": "[SEP]",
55
+ "sp_model_kwargs": {},
56
+ "tokenizer_class": "AlbertTokenizer",
57
+ "unk_token": "<unk>"
58
+ }
trainer_state.json ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 2.9333333333333336,
5
+ "eval_steps": 100,
6
+ "global_step": 66,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.4444444444444444,
13
+ "grad_norm": 24.911563873291016,
14
+ "learning_rate": 4.0000000000000003e-07,
15
+ "loss": 0.6944,
16
+ "step": 10
17
+ },
18
+ {
19
+ "epoch": 0.8888888888888888,
20
+ "grad_norm": 20.48420524597168,
21
+ "learning_rate": 8.000000000000001e-07,
22
+ "loss": 0.6317,
23
+ "step": 20
24
+ },
25
+ {
26
+ "epoch": 1.3333333333333333,
27
+ "grad_norm": 21.66866111755371,
28
+ "learning_rate": 1.2000000000000002e-06,
29
+ "loss": 0.5401,
30
+ "step": 30
31
+ },
32
+ {
33
+ "epoch": 1.7777777777777777,
34
+ "grad_norm": 12.989063262939453,
35
+ "learning_rate": 1.6000000000000001e-06,
36
+ "loss": 0.4192,
37
+ "step": 40
38
+ },
39
+ {
40
+ "epoch": 2.2222222222222223,
41
+ "grad_norm": 8.98383617401123,
42
+ "learning_rate": 2.0000000000000003e-06,
43
+ "loss": 0.29,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 2.6666666666666665,
48
+ "grad_norm": 5.708517074584961,
49
+ "learning_rate": 2.4000000000000003e-06,
50
+ "loss": 0.2052,
51
+ "step": 60
52
+ },
53
+ {
54
+ "epoch": 2.9333333333333336,
55
+ "step": 66,
56
+ "total_flos": 3605999510880.0,
57
+ "train_loss": 0.4347104240547527,
58
+ "train_runtime": 17.5364,
59
+ "train_samples_per_second": 121.12,
60
+ "train_steps_per_second": 3.764
61
+ },
62
+ {
63
+ "epoch": 2.9333333333333336,
64
+ "eval_accuracy": 1.0,
65
+ "eval_accuracy_label_bluesky": 1.0,
66
+ "eval_accuracy_label_specific": 1.0,
67
+ "eval_f1": 1.0,
68
+ "eval_loss": 0.1264917105436325,
69
+ "eval_precision": 1.0,
70
+ "eval_recall": 1.0,
71
+ "eval_runtime": 0.8403,
72
+ "eval_samples_per_second": 361.76,
73
+ "eval_steps_per_second": 22.61,
74
+ "step": 66
75
+ }
76
+ ],
77
+ "logging_steps": 10,
78
+ "max_steps": 66,
79
+ "num_input_tokens_seen": 0,
80
+ "num_train_epochs": 3,
81
+ "save_steps": 1000,
82
+ "stateful_callbacks": {
83
+ "TrainerControl": {
84
+ "args": {
85
+ "should_epoch_stop": false,
86
+ "should_evaluate": false,
87
+ "should_log": false,
88
+ "should_save": true,
89
+ "should_training_stop": true
90
+ },
91
+ "attributes": {}
92
+ }
93
+ },
94
+ "total_flos": 3605999510880.0,
95
+ "train_batch_size": 16,
96
+ "trial_name": null,
97
+ "trial_params": null
98
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:830d97dfa959466668a482c555597fa3325e36badcad0caf953958026c40a8ae
3
+ size 5176