ashishkgpian commited on
Commit
6ee4760
·
verified ·
1 Parent(s): 9bb334b

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +3 -2
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,11 +1,10 @@
1
  {
2
- "_name_or_path": "dmis-lab/biobert-v1.1",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "classifier_dropout": null,
8
- "gradient_checkpointing": false,
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
@@ -2544,10 +2543,12 @@
2544
  "layer_norm_eps": 1e-12,
2545
  "max_position_embeddings": 512,
2546
  "model_type": "bert",
 
2547
  "num_attention_heads": 12,
2548
  "num_hidden_layers": 12,
2549
  "pad_token_id": 0,
2550
  "position_embedding_type": "absolute",
 
2551
  "torch_dtype": "float32",
2552
  "transformers_version": "4.43.4",
2553
  "type_vocab_size": 2,
 
1
  {
2
+ "_name_or_path": "DATEXIS/CORe-clinical-diagnosis-prediction",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "classifier_dropout": null,
 
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
 
2543
  "layer_norm_eps": 1e-12,
2544
  "max_position_embeddings": 512,
2545
  "model_type": "bert",
2546
+ "name": "Bert",
2547
  "num_attention_heads": 12,
2548
  "num_hidden_layers": 12,
2549
  "pad_token_id": 0,
2550
  "position_embedding_type": "absolute",
2551
+ "problem_type": "multi_label_classification",
2552
  "torch_dtype": "float32",
2553
  "transformers_version": "4.43.4",
2554
  "type_vocab_size": 2,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:64526d31dd8922634e45595d0d0098e6e1e017bd4b8a115282bfc317951c6437
3
  size 437149604
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:906ee5c3a4f3605d17191ed0578e9b17d32ae0a83c6baa6e41ef44e7b50a8935
3
  size 437149604