ui-chope commited on
Commit
55037a1
1 Parent(s): cfbbdac

Upload DistilBertForTokenClassification

Browse files
Files changed (2) hide show
  1. config.json +11 -0
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -20,6 +20,17 @@
20
  "8": "I-Restaurant"
21
  },
22
  "initializer_range": 0.02,
 
 
 
 
 
 
 
 
 
 
 
23
  "max_position_embeddings": 512,
24
  "model_type": "distilbert",
25
  "n_heads": 12,
 
20
  "8": "I-Restaurant"
21
  },
22
  "initializer_range": 0.02,
23
+ "label2id": {
24
+ "B-Cuisine": 3,
25
+ "B-Dish": 5,
26
+ "B-Location": 1,
27
+ "B-Restaurant": 7,
28
+ "I-Cuisine": 4,
29
+ "I-Dish": 6,
30
+ "I-Location": 2,
31
+ "I-Restaurant": 8,
32
+ "O": 0
33
+ },
34
  "max_position_embeddings": 512,
35
  "model_type": "distilbert",
36
  "n_heads": 12,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:954208fe08408b89d6abf91b30276fbb80527188d44a522270a0d2366fd6dcce
3
  size 265512613
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:959b22d0683745606322f1a45cdf9eac833a25466964166d9bf4c89fc205388e
3
  size 265512613