SuchithTissera commited on
Commit
b5e1f6a
·
verified ·
1 Parent(s): c134a06

Upload DistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +10 -10
  2. model.safetensors +1 -1
config.json CHANGED
@@ -11,19 +11,19 @@
11
  "eos_token_id": null,
12
  "hidden_dim": 3072,
13
  "id2label": {
14
- "0": "LABEL_0",
15
- "1": "LABEL_1",
16
- "2": "LABEL_2",
17
- "3": "LABEL_3",
18
- "4": "LABEL_4"
19
  },
20
  "initializer_range": 0.02,
21
  "label2id": {
22
- "LABEL_0": 0,
23
- "LABEL_1": 1,
24
- "LABEL_2": 2,
25
- "LABEL_3": 3,
26
- "LABEL_4": 4
27
  },
28
  "max_position_embeddings": 512,
29
  "model_type": "distilbert",
 
11
  "eos_token_id": null,
12
  "hidden_dim": 3072,
13
  "id2label": {
14
+ "0": "Business",
15
+ "1": "Opinion",
16
+ "2": "Political Gossip",
17
+ "3": "Sports",
18
+ "4": "World news"
19
  },
20
  "initializer_range": 0.02,
21
  "label2id": {
22
+ "Business": 0,
23
+ "Opinion": 1,
24
+ "Political Gossip": 2,
25
+ "Sports": 3,
26
+ "World news": 4
27
  },
28
  "max_position_embeddings": 512,
29
  "model_type": "distilbert",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:990af2facd0ec94fade5b8c5650ab1ec076465874a155708cd4a74e2c42f4d46
3
  size 267841796
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c439534ed46c735c670e7f1b10e21c9cbe417a97de42a268519af24679784549
3
  size 267841796