Views
No views yet
| Model | DisTEMIST | MedProcNER | SympTEMIST |
|---|---|---|---|
| ClinLinker | 0.845 | 0.898 | 0.909 |
| ClinLinker-KB-P | 0.853 | 0.891 | 0.918 |
| ClinLinker-KB-GP | 0.864 | 0.901 | 0.922 |
| SapBERT-XLM-R-large | 0.800 | 0.850 | 0.827 |
| RoBERTa biomedical | 0.600 | 0.668 | 0.609 |
1from transformers import AutoModel, AutoTokenizer
2import torch
3
4model = AutoModel.from_pretrained("ICB-UMA/ClinLinker-KB-GP")
5tokenizer = AutoTokenizer.from_pretrained("ICB-UMA/ClinLinker-KB-GP")
6
7mention = "insuficiencia renal aguda"
8inputs = tokenizer(mention, return_tensors="pt")
9with torch.no_grad():
10 outputs = model(**inputs)
11embedding = outputs.last_hidden_state[:, 0, :]
12print(embedding.shape)FaissEncoder class.Gallego, Fernando and López-García, Guillermo and Gasco, Luis and Krallinger, Martin and Veredas, Francisco J., Clinlinker-Kb: Clinical Entity Linking in Spanish with Knowledge-Graph Enhanced Biencoders. Available at SSRN:http://dx.doi.org/10.2139/ssrn.4939986.