Hugging Face's logo

Mathildeholst
/
classifier-agnews-distilbert

Text Classification
Transformers
Safetensors
distilbert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
Community
Mathildeholst commited on
Commit
bc1fea5
·
verified ·
1 Parent(s): b7f83bd

Training in progress, step 500

Browse files
Files changed (3) hide show
  1. config.json +8 -8
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
config.json CHANGED
@@ -9,17 +9,17 @@
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2",
15
- "3": "LABEL_3"
16
  },
17
  "initializer_range": 0.02,
18
  "label2id": {
19
- "LABEL_0": 0,
20
- "LABEL_1": 1,
21
- "LABEL_2": 2,
22
- "LABEL_3": 3
23
  },
24
  "max_position_embeddings": 512,
25
  "model_type": "distilbert",
 
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "World",
13
+ "1": "Sports",
14
+ "2": "Business",
15
+ "3": "Sci/Tech"
16
  },
17
  "initializer_range": 0.02,
18
  "label2id": {
19
+ "Business": 2,
20
+ "Sci/Tech": 3,
21
+ "Sports": 1,
22
+ "World": 0
23
  },
24
  "max_position_embeddings": 512,
25
  "model_type": "distilbert",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bd367426e8a0982c94e5228ce1276493dbd86b370c7caa1212960e4c05efc240
3
  size 267838720
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc397a750c2f51b854068abc160b161f1e5930e79872c5db5c1f1513525726c8
3
  size 267838720
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b6f62798547ecf20cf6b07efcfdacc75e1048ad3459284197a83286c2ef0c719
3
  size 5777
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:787251739d20ddd8f3051c38792a79b35d29ef56bd730238e27ff1b5abbb7f58
3
  size 5777