lewtun HF staff commited on
Commit
3ba7521
1 Parent(s): f266cff

Align label mapping with imdb dataset

Browse files

Hi there, your model is using a default label mapping. Accept this PR to align the label mapping with the imdb dataset this model was trained on. This will enable your model to be evaluated by [Hugging Face's automatic model evaluator](https://huggingface.co/spaces/autoevaluate/model-evaluator?dataset=imdb)

Files changed (1) hide show
  1. config.json +4 -4
config.json CHANGED
@@ -9,13 +9,13 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1"
14
  },
15
  "initializer_range": 0.02,
16
  "label2id": {
17
- "LABEL_0": 0,
18
- "LABEL_1": 1
19
  },
20
  "max_position_embeddings": 512,
21
  "model_type": "distilbert",
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "neg",
13
+ "1": "pos"
14
  },
15
  "initializer_range": 0.02,
16
  "label2id": {
17
+ "neg": 0,
18
+ "pos": 1
19
  },
20
  "max_position_embeddings": 512,
21
  "model_type": "distilbert",