Views
No views yet
| Evaluation Split | Top-25 Accuracy |
|---|---|
| Gold Standard | 0.917 |
| Unseen Mentions | 0.831 |
| Unseen Codes | 0.808 |
1from transformers import AutoModel, AutoTokenizer
2import torch
3
4model = AutoModel.from_pretrained("ICB-UMA/MedProcNER-bi-encoder")
5tokenizer = AutoTokenizer.from_pretrained("ICB-UMA/MedProcNER-bi-encoder")
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 for efficient retrieval.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