Kanit commited on
Commit
c1d28ea
1 Parent(s): 50cf9b8

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +8 -0
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -14,8 +14,16 @@
14
  "hidden_act": "gelu",
15
  "hidden_dropout_prob": 0.1,
16
  "hidden_size": 128,
 
 
 
 
17
  "initializer_range": 0.02,
18
  "intermediate_size": 512,
 
 
 
 
19
  "layer_norm_eps": 1e-12,
20
  "max_position_embeddings": 512,
21
  "model_type": "bert",
 
14
  "hidden_act": "gelu",
15
  "hidden_dropout_prob": 0.1,
16
  "hidden_size": 128,
17
+ "id2label": {
18
+ "0": "non-toxic",
19
+ "1": "toxic"
20
+ },
21
  "initializer_range": 0.02,
22
  "intermediate_size": 512,
23
+ "label2id": {
24
+ "non-toxic": 0,
25
+ "toxic": 1
26
+ },
27
  "layer_norm_eps": 1e-12,
28
  "max_position_embeddings": 512,
29
  "model_type": "bert",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:620aa9adc774a2b01a068b98163713314eff61a3bc55578d9da09f43d3cd0fb2
3
  size 17562607
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a87ce74f3d35f4c1c01f14e9bd08c0cecb07526bd0ba80edea64b68100f4565
3
  size 17562607