Hugging Face's logo

nlpie
/
clinical-mobilebert-i2b2-2010

Token Classification
Transformers
PyTorch
mobilebert
Inference Endpoints
Model card Files Community
4
mohammadmahdinouri commited on
Commit
c02a0f0
·
1 Parent(s): b0414b3

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -7
config.json CHANGED
@@ -24,13 +24,13 @@
24
  "intra_bottleneck_size": 128,
25
  "key_query_shared_bottleneck": true,
26
  "label2id": {
27
- "LABEL_1": 0,
28
- "LABEL_2": 1,
29
- "LABEL_3": 2,
30
- "LABEL_4": 3,
31
- "LABEL_5": 4,
32
- "LABEL_6": 5,
33
- "LABEL_7": 6
34
  },
35
  "layer_norm_eps": 1e-12,
36
  "max_position_embeddings": 512,
 
24
  "intra_bottleneck_size": 128,
25
  "key_query_shared_bottleneck": true,
26
  "label2id": {
27
+ "I-treatment": 0,
28
+ "O": 1,
29
+ "B-test": 2,
30
+ "I-problem": 3,
31
+ "B-treatment": 4,
32
+ "I-test": 5,
33
+ "B-problem": 6
34
  },
35
  "layer_norm_eps": 1e-12,
36
  "max_position_embeddings": 512,