Upload BertForSequenceClassification

README.md

@@ -1,4 +1,7 @@
 ---
+language:
+- en
+license: apache-2.0
 library_name: transformers
 tags:
 - biomedical
@@ -7,9 +10,6 @@ tags:
 - genetics
 - bert
 - fine-tuned
-language:
-- en
-license: apache-2.0
 base_model: dmis-lab/biobert-large-cased-v1.1
 datasets:
 - clinvar

config.json

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "experiment/conclusion_description_removed_data/trained_models/2024-06-16/biobert-v1.1_16-03-42",
   "architectures": [
-    "BertLMHeadModel"
+    "BertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
@@ -10,17 +10,17 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "P/LP",
+    "0": "B/LB",
     "1": "VUS",
-    "2": "B/LB"
+    "2": "P/LP"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "P/LP": 0,
-    "VUS": 1,
-    "B/LB": 2
-  }
+    "B/LB": 0,
+    "P/LP": 2,
+    "VUS": 1
+  },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",
@@ -30,7 +30,7 @@
   "position_embedding_type": "absolute",
   "problem_type": "single_label_classification",
   "torch_dtype": "float32",
-  "transformers_version": "4.46.3",
+  "transformers_version": "4.40.2",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 28996

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fc848a01e3f97adf521ad3d35e808bad8420755bf978c1be192aac93ffc1f02f
-size 433386912
+oid sha256:0b8bbf3cc9a95a81474ba2c3b00d2599e5e15b4d5c2ee66c7557e00336dddea2
+size 433273844