Hugging Face's logo

dima806
/
news-category-classifier-distilbert

Text Classification
Transformers
PyTorch
Safetensors
bert
Inference Endpoints
Model card Files Community
1
dima806 commited on
Commit
5c47c9b
•
1 Parent(s): 700eeae

Upload folder using huggingface_hub

Browse files
Files changed (11) hide show
  1. checkpoint-26190/config.json +115 -0
  2. checkpoint-26190/optimizer.pt +3 -0
  3. checkpoint-26190/pytorch_model.bin +3 -0
  4. checkpoint-26190/rng_state.pth +3 -0
  5. checkpoint-26190/scheduler.pt +3 -0
  6. checkpoint-26190/trainer_state.json +0 -0
  7. checkpoint-26190/training_args.bin +3 -0
  8. config.json +115 -0
  9. pytorch_model.bin +3 -0
  10. training_args.bin +3 -0
  11. vocab.txt +0 -0
checkpoint-26190/config.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-uncased",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "ARTS",
14
+ "1": "ARTS & CULTURE",
15
+ "2": "BLACK VOICES",
16
+ "3": "BUSINESS",
17
+ "4": "COLLEGE",
18
+ "5": "COMEDY",
19
+ "6": "CRIME",
20
+ "7": "CULTURE & ARTS",
21
+ "8": "DIVORCE",
22
+ "9": "EDUCATION",
23
+ "10": "ENTERTAINMENT",
24
+ "11": "ENVIRONMENT",
25
+ "12": "FIFTY",
26
+ "13": "FOOD & DRINK",
27
+ "14": "GOOD NEWS",
28
+ "15": "GREEN",
29
+ "16": "HEALTHY LIVING",
30
+ "17": "HOME & LIVING",
31
+ "18": "IMPACT",
32
+ "19": "LATINO VOICES",
33
+ "20": "MEDIA",
34
+ "21": "MONEY",
35
+ "22": "PARENTING",
36
+ "23": "PARENTS",
37
+ "24": "POLITICS",
38
+ "25": "QUEER VOICES",
39
+ "26": "RELIGION",
40
+ "27": "SCIENCE",
41
+ "28": "SPORTS",
42
+ "29": "STYLE",
43
+ "30": "STYLE & BEAUTY",
44
+ "31": "TASTE",
45
+ "32": "TECH",
46
+ "33": "THE WORLDPOST",
47
+ "34": "TRAVEL",
48
+ "35": "U.S. NEWS",
49
+ "36": "WEDDINGS",
50
+ "37": "WEIRD NEWS",
51
+ "38": "WELLNESS",
52
+ "39": "WOMEN",
53
+ "40": "WORLD NEWS",
54
+ "41": "WORLDPOST"
55
+ },
56
+ "initializer_range": 0.02,
57
+ "intermediate_size": 3072,
58
+ "label2id": {
59
+ "ARTS": 0,
60
+ "ARTS & CULTURE": 1,
61
+ "BLACK VOICES": 2,
62
+ "BUSINESS": 3,
63
+ "COLLEGE": 4,
64
+ "COMEDY": 5,
65
+ "CRIME": 6,
66
+ "CULTURE & ARTS": 7,
67
+ "DIVORCE": 8,
68
+ "EDUCATION": 9,
69
+ "ENTERTAINMENT": 10,
70
+ "ENVIRONMENT": 11,
71
+ "FIFTY": 12,
72
+ "FOOD & DRINK": 13,
73
+ "GOOD NEWS": 14,
74
+ "GREEN": 15,
75
+ "HEALTHY LIVING": 16,
76
+ "HOME & LIVING": 17,
77
+ "IMPACT": 18,
78
+ "LATINO VOICES": 19,
79
+ "MEDIA": 20,
80
+ "MONEY": 21,
81
+ "PARENTING": 22,
82
+ "PARENTS": 23,
83
+ "POLITICS": 24,
84
+ "QUEER VOICES": 25,
85
+ "RELIGION": 26,
86
+ "SCIENCE": 27,
87
+ "SPORTS": 28,
88
+ "STYLE": 29,
89
+ "STYLE & BEAUTY": 30,
90
+ "TASTE": 31,
91
+ "TECH": 32,
92
+ "THE WORLDPOST": 33,
93
+ "TRAVEL": 34,
94
+ "U.S. NEWS": 35,
95
+ "WEDDINGS": 36,
96
+ "WEIRD NEWS": 37,
97
+ "WELLNESS": 38,
98
+ "WOMEN": 39,
99
+ "WORLD NEWS": 40,
100
+ "WORLDPOST": 41
101
+ },
102
+ "layer_norm_eps": 1e-12,
103
+ "max_position_embeddings": 512,
104
+ "model_type": "bert",
105
+ "num_attention_heads": 12,
106
+ "num_hidden_layers": 12,
107
+ "pad_token_id": 0,
108
+ "position_embedding_type": "absolute",
109
+ "problem_type": "single_label_classification",
110
+ "torch_dtype": "float32",
111
+ "transformers_version": "4.33.0",
112
+ "type_vocab_size": 2,
113
+ "use_cache": true,
114
+ "vocab_size": 30522
115
+ }
checkpoint-26190/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9536db4a7803eb7c6eb5641980f4560a8f2e396a6d4080c5d91840c920c3e42d
3
+ size 876283909
checkpoint-26190/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8acbe5c0f0fb9d25be4f9f33ec6539eaa507207a1c6391ad83472afe8594d9cd
3
+ size 438126513
checkpoint-26190/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8429a174ba2557c2599455ba36c20ff5eb647d30f8320d7a833ae64759585045
3
+ size 14575
checkpoint-26190/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5fad7781424ce42cdc4a82945e33af92ea43e6b9e6337de998f4aa1397e5775
3
+ size 627
checkpoint-26190/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-26190/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92affd5b51510b928ccf9a7d9064a62bd7796ca3d2668f003694a847e0ff17b7
3
+ size 4027
config.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-uncased",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "ARTS",
14
+ "1": "ARTS & CULTURE",
15
+ "2": "BLACK VOICES",
16
+ "3": "BUSINESS",
17
+ "4": "COLLEGE",
18
+ "5": "COMEDY",
19
+ "6": "CRIME",
20
+ "7": "CULTURE & ARTS",
21
+ "8": "DIVORCE",
22
+ "9": "EDUCATION",
23
+ "10": "ENTERTAINMENT",
24
+ "11": "ENVIRONMENT",
25
+ "12": "FIFTY",
26
+ "13": "FOOD & DRINK",
27
+ "14": "GOOD NEWS",
28
+ "15": "GREEN",
29
+ "16": "HEALTHY LIVING",
30
+ "17": "HOME & LIVING",
31
+ "18": "IMPACT",
32
+ "19": "LATINO VOICES",
33
+ "20": "MEDIA",
34
+ "21": "MONEY",
35
+ "22": "PARENTING",
36
+ "23": "PARENTS",
37
+ "24": "POLITICS",
38
+ "25": "QUEER VOICES",
39
+ "26": "RELIGION",
40
+ "27": "SCIENCE",
41
+ "28": "SPORTS",
42
+ "29": "STYLE",
43
+ "30": "STYLE & BEAUTY",
44
+ "31": "TASTE",
45
+ "32": "TECH",
46
+ "33": "THE WORLDPOST",
47
+ "34": "TRAVEL",
48
+ "35": "U.S. NEWS",
49
+ "36": "WEDDINGS",
50
+ "37": "WEIRD NEWS",
51
+ "38": "WELLNESS",
52
+ "39": "WOMEN",
53
+ "40": "WORLD NEWS",
54
+ "41": "WORLDPOST"
55
+ },
56
+ "initializer_range": 0.02,
57
+ "intermediate_size": 3072,
58
+ "label2id": {
59
+ "ARTS": 0,
60
+ "ARTS & CULTURE": 1,
61
+ "BLACK VOICES": 2,
62
+ "BUSINESS": 3,
63
+ "COLLEGE": 4,
64
+ "COMEDY": 5,
65
+ "CRIME": 6,
66
+ "CULTURE & ARTS": 7,
67
+ "DIVORCE": 8,
68
+ "EDUCATION": 9,
69
+ "ENTERTAINMENT": 10,
70
+ "ENVIRONMENT": 11,
71
+ "FIFTY": 12,
72
+ "FOOD & DRINK": 13,
73
+ "GOOD NEWS": 14,
74
+ "GREEN": 15,
75
+ "HEALTHY LIVING": 16,
76
+ "HOME & LIVING": 17,
77
+ "IMPACT": 18,
78
+ "LATINO VOICES": 19,
79
+ "MEDIA": 20,
80
+ "MONEY": 21,
81
+ "PARENTING": 22,
82
+ "PARENTS": 23,
83
+ "POLITICS": 24,
84
+ "QUEER VOICES": 25,
85
+ "RELIGION": 26,
86
+ "SCIENCE": 27,
87
+ "SPORTS": 28,
88
+ "STYLE": 29,
89
+ "STYLE & BEAUTY": 30,
90
+ "TASTE": 31,
91
+ "TECH": 32,
92
+ "THE WORLDPOST": 33,
93
+ "TRAVEL": 34,
94
+ "U.S. NEWS": 35,
95
+ "WEDDINGS": 36,
96
+ "WEIRD NEWS": 37,
97
+ "WELLNESS": 38,
98
+ "WOMEN": 39,
99
+ "WORLD NEWS": 40,
100
+ "WORLDPOST": 41
101
+ },
102
+ "layer_norm_eps": 1e-12,
103
+ "max_position_embeddings": 512,
104
+ "model_type": "bert",
105
+ "num_attention_heads": 12,
106
+ "num_hidden_layers": 12,
107
+ "pad_token_id": 0,
108
+ "position_embedding_type": "absolute",
109
+ "problem_type": "single_label_classification",
110
+ "torch_dtype": "float32",
111
+ "transformers_version": "4.33.0",
112
+ "type_vocab_size": 2,
113
+ "use_cache": true,
114
+ "vocab_size": 30522
115
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8acbe5c0f0fb9d25be4f9f33ec6539eaa507207a1c6391ad83472afe8594d9cd
3
+ size 438126513
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92affd5b51510b928ccf9a7d9064a62bd7796ca3d2668f003694a847e0ff17b7
3
+ size 4027
vocab.txt ADDED
The diff for this file is too large to render. See raw diff