Text Classification
Transformers
Safetensors
modernbert
biomedical
literature-mining
genomics
rare-disease
text-embeddings-inference
Instructions to use tmy100000001/LitDD_BERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use tmy100000001/LitDD_BERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="tmy100000001/LitDD_BERT")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("tmy100000001/LitDD_BERT") model = AutoModelForSequenceClassification.from_pretrained("tmy100000001/LitDD_BERT", device_map="auto") - Notebooks
- Google Colab
- Kaggle
add20k retrain (seed 44): 20k corpus negatives, 24.0% train prevalence; PubMed fire rate 19.46% -> 0.51%; test F1 0.914
Browse files- config.json +10 -10
- model.safetensors +1 -1
config.json
CHANGED
|
@@ -20,9 +20,17 @@
|
|
| 20 |
"gradient_checkpointing": false,
|
| 21 |
"hidden_activation": "gelu",
|
| 22 |
"hidden_size": 1024,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
"initializer_cutoff_factor": 2.0,
|
| 24 |
"initializer_range": 0.02,
|
| 25 |
"intermediate_size": 2624,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
"layer_norm_eps": 1e-05,
|
| 27 |
"local_attention": 128,
|
| 28 |
"local_rope_theta": 10000.0,
|
|
@@ -44,13 +52,5 @@
|
|
| 44 |
"sparse_prediction": false,
|
| 45 |
"torch_dtype": "float32",
|
| 46 |
"transformers_version": "4.48.3",
|
| 47 |
-
"vocab_size": 50368
|
| 48 |
-
|
| 49 |
-
"0": "NOT_RELEVANT",
|
| 50 |
-
"1": "RELEVANT"
|
| 51 |
-
},
|
| 52 |
-
"label2id": {
|
| 53 |
-
"NOT_RELEVANT": 0,
|
| 54 |
-
"RELEVANT": 1
|
| 55 |
-
}
|
| 56 |
-
}
|
|
|
|
| 20 |
"gradient_checkpointing": false,
|
| 21 |
"hidden_activation": "gelu",
|
| 22 |
"hidden_size": 1024,
|
| 23 |
+
"id2label": {
|
| 24 |
+
"0": "NOT_RELEVANT",
|
| 25 |
+
"1": "RELEVANT"
|
| 26 |
+
},
|
| 27 |
"initializer_cutoff_factor": 2.0,
|
| 28 |
"initializer_range": 0.02,
|
| 29 |
"intermediate_size": 2624,
|
| 30 |
+
"label2id": {
|
| 31 |
+
"NOT_RELEVANT": 0,
|
| 32 |
+
"RELEVANT": 1
|
| 33 |
+
},
|
| 34 |
"layer_norm_eps": 1e-05,
|
| 35 |
"local_attention": 128,
|
| 36 |
"local_rope_theta": 10000.0,
|
|
|
|
| 52 |
"sparse_prediction": false,
|
| 53 |
"torch_dtype": "float32",
|
| 54 |
"transformers_version": "4.48.3",
|
| 55 |
+
"vocab_size": 50368
|
| 56 |
+
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1583351632
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b868b0bbd123983036685ffb595d00a3220f1b096bd918e69884f2c7a4a9c5f
|
| 3 |
size 1583351632
|