Rostlab
/

prot_bert_bfd_membrane

Text Classification

Model card Files Files and versions

system HF Staff commited on Oct 7, 2020

Commit

bbd567e

·

1 Parent(s): 2a8eff5

Update config.json

Files changed (1) hide show

config.json +10 -1

config.json CHANGED Viewed

@@ -16,5 +16,14 @@
   "num_hidden_layers": 30,
   "pad_token_id": 0,
   "type_vocab_size": 2,
-  "vocab_size": 30
 }

   "num_hidden_layers": 30,
   "pad_token_id": 0,
   "type_vocab_size": 2,
+  "vocab_size": 30,
+  "id2label": {
+    "0": "Soluble",
+    "1": "Membrane"
+  },
+  "initializer_range": 0.02,
+  "label2id": {
+    "Soluble": 0,
+    "Membrane": 1
+  },
 }