Hugging Face's logo

Hemg
/
EMOTION-AI

Text Classification
Transformers
Safetensors
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Community
Hemg commited on
Commit
2fe627c
·
verified ·
1 Parent(s): 2498367

Training in progress, epoch 3

Browse files
Files changed (5) hide show
  1. config.json +85 -85
  2. model.safetensors +1 -1
  3. special_tokens_map.json +7 -7
  4. tokenizer_config.json +55 -56
  5. training_args.bin +2 -2
config.json CHANGED
@@ -1,85 +1,85 @@
1
- {
2
- "_name_or_path": "distilbert/distilbert-base-uncased",
3
- "activation": "gelu",
4
- "architectures": [
5
- "DistilBertForSequenceClassification"
6
- ],
7
- "attention_dropout": 0.1,
8
- "dim": 768,
9
- "dropout": 0.1,
10
- "hidden_dim": 3072,
11
- "id2label": {
12
- "0": "admiration",
13
- "1": "amusement",
14
- "2": "anger",
15
- "3": "annoyance",
16
- "4": "approval",
17
- "5": "caring",
18
- "6": "confusion",
19
- "7": "curiosity",
20
- "8": "desire",
21
- "9": "disappointment",
22
- "10": "disapproval",
23
- "11": "disgust",
24
- "12": "embarrassment",
25
- "13": "excitement",
26
- "14": "fear",
27
- "15": "gratitude",
28
- "16": "grief",
29
- "17": "joy",
30
- "18": "love",
31
- "19": "nervousness",
32
- "20": "optimism",
33
- "21": "pride",
34
- "22": "realization",
35
- "23": "relief",
36
- "24": "remorse",
37
- "25": "sadness",
38
- "26": "surprise",
39
- "27": "neutral"
40
- },
41
- "initializer_range": 0.02,
42
- "label2id": {
43
- "admiration": 0,
44
- "amusement": 1,
45
- "anger": 2,
46
- "annoyance": 3,
47
- "approval": 4,
48
- "caring": 5,
49
- "confusion": 6,
50
- "curiosity": 7,
51
- "desire": 8,
52
- "disappointment": 9,
53
- "disapproval": 10,
54
- "disgust": 11,
55
- "embarrassment": 12,
56
- "excitement": 13,
57
- "fear": 14,
58
- "gratitude": 15,
59
- "grief": 16,
60
- "joy": 17,
61
- "love": 18,
62
- "nervousness": 19,
63
- "neutral": 27,
64
- "optimism": 20,
65
- "pride": 21,
66
- "realization": 22,
67
- "relief": 23,
68
- "remorse": 24,
69
- "sadness": 25,
70
- "surprise": 26
71
- },
72
- "max_position_embeddings": 512,
73
- "model_type": "distilbert",
74
- "n_heads": 12,
75
- "n_layers": 6,
76
- "pad_token_id": 0,
77
- "problem_type": "single_label_classification",
78
- "qa_dropout": 0.1,
79
- "seq_classif_dropout": 0.2,
80
- "sinusoidal_pos_embds": false,
81
- "tie_weights_": true,
82
- "torch_dtype": "float32",
83
- "transformers_version": "4.47.1",
84
- "vocab_size": 30522
85
- }
 
1
+ {
2
+ "_name_or_path": "distilbert/distilbert-base-uncased",
3
+ "activation": "gelu",
4
+ "architectures": [
5
+ "DistilBertForSequenceClassification"
6
+ ],
7
+ "attention_dropout": 0.1,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "admiration",
13
+ "1": "amusement",
14
+ "2": "anger",
15
+ "3": "annoyance",
16
+ "4": "approval",
17
+ "5": "caring",
18
+ "6": "confusion",
19
+ "7": "curiosity",
20
+ "8": "desire",
21
+ "9": "disappointment",
22
+ "10": "disapproval",
23
+ "11": "disgust",
24
+ "12": "embarrassment",
25
+ "13": "excitement",
26
+ "14": "fear",
27
+ "15": "gratitude",
28
+ "16": "grief",
29
+ "17": "joy",
30
+ "18": "love",
31
+ "19": "nervousness",
32
+ "20": "optimism",
33
+ "21": "pride",
34
+ "22": "realization",
35
+ "23": "relief",
36
+ "24": "remorse",
37
+ "25": "sadness",
38
+ "26": "surprise",
39
+ "27": "neutral"
40
+ },
41
+ "initializer_range": 0.02,
42
+ "label2id": {
43
+ "admiration": 0,
44
+ "amusement": 1,
45
+ "anger": 2,
46
+ "annoyance": 3,
47
+ "approval": 4,
48
+ "caring": 5,
49
+ "confusion": 6,
50
+ "curiosity": 7,
51
+ "desire": 8,
52
+ "disappointment": 9,
53
+ "disapproval": 10,
54
+ "disgust": 11,
55
+ "embarrassment": 12,
56
+ "excitement": 13,
57
+ "fear": 14,
58
+ "gratitude": 15,
59
+ "grief": 16,
60
+ "joy": 17,
61
+ "love": 18,
62
+ "nervousness": 19,
63
+ "neutral": 27,
64
+ "optimism": 20,
65
+ "pride": 21,
66
+ "realization": 22,
67
+ "relief": 23,
68
+ "remorse": 24,
69
+ "sadness": 25,
70
+ "surprise": 26
71
+ },
72
+ "max_position_embeddings": 512,
73
+ "model_type": "distilbert",
74
+ "n_heads": 12,
75
+ "n_layers": 6,
76
+ "pad_token_id": 0,
77
+ "problem_type": "single_label_classification",
78
+ "qa_dropout": 0.1,
79
+ "seq_classif_dropout": 0.2,
80
+ "sinusoidal_pos_embds": false,
81
+ "tie_weights_": true,
82
+ "torch_dtype": "float32",
83
+ "transformers_version": "4.44.2",
84
+ "vocab_size": 30522
85
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a962d51a1ccf7b3cecb57cce1a2aa1c3e508c6f29fc9da8e8c5185fd797ec987
3
  size 267912544
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93ce520b624564ab8bb11c43a76ed961ea1a15f912b5c1cdaec062efb33113b6
3
  size 267912544
special_tokens_map.json CHANGED
@@ -1,7 +1,7 @@
1
- {
2
- "cls_token": "[CLS]",
3
- "mask_token": "[MASK]",
4
- "pad_token": "[PAD]",
5
- "sep_token": "[SEP]",
6
- "unk_token": "[UNK]"
7
- }
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer_config.json CHANGED
@@ -1,56 +1,55 @@
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
- "100": {
12
- "content": "[UNK]",
13
- "lstrip": false,
14
- "normalized": false,
15
- "rstrip": false,
16
- "single_word": false,
17
- "special": true
18
- },
19
- "101": {
20
- "content": "[CLS]",
21
- "lstrip": false,
22
- "normalized": false,
23
- "rstrip": false,
24
- "single_word": false,
25
- "special": true
26
- },
27
- "102": {
28
- "content": "[SEP]",
29
- "lstrip": false,
30
- "normalized": false,
31
- "rstrip": false,
32
- "single_word": false,
33
- "special": true
34
- },
35
- "103": {
36
- "content": "[MASK]",
37
- "lstrip": false,
38
- "normalized": false,
39
- "rstrip": false,
40
- "single_word": false,
41
- "special": true
42
- }
43
- },
44
- "clean_up_tokenization_spaces": false,
45
- "cls_token": "[CLS]",
46
- "do_lower_case": true,
47
- "extra_special_tokens": {},
48
- "mask_token": "[MASK]",
49
- "model_max_length": 512,
50
- "pad_token": "[PAD]",
51
- "sep_token": "[SEP]",
52
- "strip_accents": null,
53
- "tokenize_chinese_chars": true,
54
- "tokenizer_class": "DistilBertTokenizer",
55
- "unk_token": "[UNK]"
56
- }
 
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
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_lower_case": true,
47
+ "mask_token": "[MASK]",
48
+ "model_max_length": 512,
49
+ "pad_token": "[PAD]",
50
+ "sep_token": "[SEP]",
51
+ "strip_accents": null,
52
+ "tokenize_chinese_chars": true,
53
+ "tokenizer_class": "DistilBertTokenizer",
54
+ "unk_token": "[UNK]"
55
+ }
 
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c4008b92b1ad4c6361c68acef4c4bd6c8091fea664c788251208b15741507573
3
- size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89c818e211038864cd7efefd05c1584ba4bd8f1966948092bef088ee70cae885
3
+ size 5176