ML Wong commited on
Commit
082f536
1 Parent(s): 2c435d9

Update classifier to have a Unknown class prediction

Browse files
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ *__pycache__*
2
+ *pyc
models/npc-bert-cls/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "npc_bert-best",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
@@ -10,13 +10,15 @@
10
  "hidden_size": 768,
11
  "id2label": {
12
  "0": "Benign/Healthy",
13
- "1": "NPC"
 
14
  },
15
  "initializer_range": 0.02,
16
  "intermediate_size": 3072,
17
  "label2id": {
18
  "Benign/Healthy": 0,
19
- "NPC": 1
 
20
  },
21
  "layer_norm_eps": 1e-12,
22
  "max_position_embeddings": 512,
 
1
  {
2
+ "_name_or_path": "gradio_demo/models/npc-bert-best/",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
 
10
  "hidden_size": 768,
11
  "id2label": {
12
  "0": "Benign/Healthy",
13
+ "1": "NPC",
14
+ "2": "Unknown/Out of Domain"
15
  },
16
  "initializer_range": 0.02,
17
  "intermediate_size": 3072,
18
  "label2id": {
19
  "Benign/Healthy": 0,
20
+ "NPC": 1,
21
+ "Unknown/Out of Domain": 2
22
  },
23
  "layer_norm_eps": 1e-12,
24
  "max_position_embeddings": 512,
models/npc-bert-cls/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e0b76162187e0bc827c6178cb1351c15bb95a588d371028bab5c532de1d26e0f
3
- size 437958648
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4dfdc98e5c5ae51ec50742ae026b18e73540b58c6a465a83c9ca36525edaa710
3
+ size 437961724
models/npc-bert-cls/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d21e77960bd0b33c5de1bf3b3173e81dd975fb8dc083a8f1359bedf017d94d9b
3
- size 4207
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a0a0352080ce09549833a55f2979717a54dcc659c5f85b7b87a47b5eaa9d6cc
3
+ size 4664