Hugging Face's logo

valurank
/
distilbert-quality

Text Classification
Transformers
PyTorch
English
distilbert
Inference Endpoints
Model card Files Community
1
ytsaig commited on
Commit
bab4ff9
·
1 Parent(s): 3c79cd3

Add class labels.

Browse files
Files changed (3) hide show
  1. config.json +6 -6
  2. pytorch_model.bin +1 -1
  3. training_args.bin +1 -1
config.json CHANGED
@@ -9,15 +9,15 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2"
15
  },
16
  "initializer_range": 0.02,
17
  "label2id": {
18
- "LABEL_0": 0,
19
- "LABEL_1": 1,
20
- "LABEL_2": 2
21
  },
22
  "max_position_embeddings": 512,
23
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "bad",
13
+ "1": "medium",
14
+ "2": "good"
15
  },
16
  "initializer_range": 0.02,
17
  "label2id": {
18
+ "bad": 0,
19
+ "good": 2,
20
+ "medium": 1
21
  },
22
  "max_position_embeddings": 512,
23
  "model_type": "distilbert",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:097bc77ea19d0701e39f459b88efe3e9ba79a3bbc709a7ba77710086e4a5ac65
3
  size 267866263
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1af3696de76ea85a6acff551d67eaa737d081b1e140042b3bdce43bbe732b66f
3
  size 267866263
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:994e95a096d91eedc81118ccd3cd639026c5bdef6dccf75a41d86d4c15651999
3
  size 2863
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4bbd03a7afa92f3f12081e2d42e11ad663f54ff242839c87bf2d2e5829fe554c
3
  size 2863